Spaces:

fffiloni
/

imagic-stable-diffusion

Paused

fffiloni commited on Nov 25, 2022

Commit

74ec699

•

1 Parent(s): e2852a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ pipe = DiffusionPipeline.from_pretrained(
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)
-generator = th.Generator("cuda").manual_seed(0)
 def infer(prompt, init_image):
     res = pipe.train(

     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)
+generator = torch.Generator("cuda").manual_seed(0)
 def infer(prompt, init_image):
     res = pipe.train(