evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Note about cogview

Browse Source
This commit is contained in:
Jaret Burkett committed 2025-03-05 14:46:11 -07:00
1 parent 4fe33f51c1
commit 763128ea42
1 file changed
+3
Download Patch File Download Diff File Expand all files Collapse all files
toolkit/models/cogview4.py
+3
View File
@@ -1,3 +1,6 @@
# DONT USE THIS!. IT DOES NOT WORK YET!
# Will revisit this when they release more info on how it was trained.
import weakref
from diffusers import CogView4Pipeline
import torch