Upload app.py

app.py

@@ -256,7 +256,7 @@ def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, ca
                             scheduler=noise_scheduler,
                             safety_checker=StableDiffusionSafetyChecker,
                             feature_extractor=CLIPImageProcessor)
-    generator = torch.Generator(device).manual_seed(seed) if seed is not None else None
+    generator = torch.Generator(args.device).manual_seed(seed) if seed is not None else None
     output = pipe(
         ref_image=vae_clothes,
         prompt=prompt,