Update app.py

app.py

@@ -20,7 +20,6 @@ from huggingface_hub import snapshot_download
 #global tokenizer
 #global noise_scheduler
 device = "cuda:0"
-generator = torch.Generator(device=device)
 
 models_path = snapshot_download(repo_id="Snapchat/w2w")
 
@@ -41,7 +40,7 @@ network = sample_weights(unet, proj, mean, std, v[:, :1000], device, factor = 1.
 #    global network
 #    unet, _, _, _, _ = load_models(device)
 
-def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
+def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     #global device
     #global generator 
     #global unet
@@ -49,7 +48,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     #global text_encoder
     #global tokenizer
     #global noise_scheduler
-    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,