diff --git a/toolkit/models/cogview4.py b/toolkit/models/cogview4.py index 62af5498..593fa977 100644 --- a/toolkit/models/cogview4.py +++ b/toolkit/models/cogview4.py @@ -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