Spaces:

jingyangcarl
/

matgen

Sleeping

jingyangcarl commited on 27 days ago

Commit

e336d0c

1 Parent(s): 669037e

debug zerogpu

Files changed (1) hide show

model.py CHANGED Viewed

@@ -121,7 +121,7 @@ class Model:
         seed: int,
     ) -> list[PIL.Image.Image]:
         generator = torch.Generator().manual_seed(seed)
-        self.pipe.to(self.device)
         return self.pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,

         seed: int,
     ) -> list[PIL.Image.Image]:
         generator = torch.Generator().manual_seed(seed)
+        # self.pipe.to(self.device)
         return self.pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,