Update app.py

app.py

@@ -65,7 +65,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     global text_encoder
     global tokenizer
     global noise_scheduler
-    torch.Generator(device=device).manual_seed(seed)
+    generator = torch.Generator(device=device).manual_seed(seed)
     latents = torch.randn(
         (1, unet.in_channels, 512 // 8, 512 // 8),
         generator = generator,
@@ -111,7 +111,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
 def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
 
     global device
-    global generator 
+    #global generator 
     global unet
     global vae 
     global text_encoder
@@ -136,7 +136,7 @@ def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, st
 
     edited_weights = original_weights+a1*1e6*young_pad+a2*1e6*pointy_pad+a3*1e6*wavy_pad+a4*2e6*thick_pad
    
-    generator = generator.manual_seed(seed)
+    generator = torch.Generator(device=device).manual_seed(seed)
     latents = torch.randn(
         (1, unet.in_channels, 512 // 8, 512 // 8),
         generator = generator,