ai-lab-pipeline-video-image

Runtime error

kadirnar commited on Feb 13

Commit

7e1ec9f

•

1 Parent(s): d1b6cf8

Update stable_cascade.py

Files changed (1) hide show

stable_cascade.py CHANGED Viewed

@@ -31,7 +31,7 @@ def generate_images(
     Returns:
     - List[PIL.Image]: A list of generated PIL Image objects.
     """
-    generator = torch.Generator(device="cuda").manual_seed(seed)
     # Generate image embeddings using the prior model
     prior_output = prior(

     Returns:
     - List[PIL.Image]: A list of generated PIL Image objects.
     """
+    generator = torch.Generator(device="cuda").manual_seed(int(seed))
     # Generate image embeddings using the prior model
     prior_output = prior(