Spaces:

yizhezhu
/

MoMA_zeroGPU

Running on Zero

Kunpeng Song commited on Jun 6, 2024

Commit

57f8019

1 Parent(s): ec3730d

fix zero

Files changed (2) hide show

.DS_Store CHANGED Viewed

Binary files a/.DS_Store and b/.DS_Store differ

model_lib/moMA_generator.py CHANGED Viewed

@@ -95,7 +95,7 @@ class MoMA_generator:
             vae=vae,
             feature_extractor=None,
             safety_checker=None,
-        ).to(self.device)
         self.unet = self.pipe.unet
         add_function(self.pipe)

             vae=vae,
             feature_extractor=None,
             safety_checker=None,
+        )
         self.unet = self.pipe.unet
         add_function(self.pipe)