Spaces:

CompVis
/

celeba-latent-diffusion

Runtime error

osanseviero HF staff commited on Jul 18, 2022

Commit

ea59a12

•

1 Parent(s): 7854755

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ pipeline = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusi
 def predict(seed):
-    torch.manual_seed(seed)
     image = pipeline(generator=generator, num_inference_steps=1)["sample"]
     image_processed = image.cpu().permute(0, 2, 3, 1)
     image_processed = (image_processed + 1.0) * 127.5

 def predict(seed):
+    generator = torch.manual_seed(seed)
     image = pipeline(generator=generator, num_inference_steps=1)["sample"]
     image_processed = image.cpu().permute(0, 2, 3, 1)
     image_processed = (image_processed + 1.0) * 127.5