Spaces:

Pie31415
/

control-animation

Build error

Pie31415 commited on May 2, 2023

Commit

d158f74

1 Parent(s): ac67678

Update text_to_animation/model.py

Files changed (1) hide show

text_to_animation/model.py CHANGED Viewed

@@ -404,7 +404,7 @@ class ControlAnimationModel:
         cfg_scale: float = 7.0,
         seed: int = 0,
     ) -> List[Image.Image]:
-        generator = torch.Generator(device=self.device).manual_seed(seed)
         pipe = StableDiffusionPipeline.from_pretrained(model_link)
         batch_size = 4

         cfg_scale: float = 7.0,
         seed: int = 0,
     ) -> List[Image.Image]:
+        generator = torch.Generator().manual_seed(seed)
         pipe = StableDiffusionPipeline.from_pretrained(model_link)
         batch_size = 4