Upload app.py

app.py

@@ -76,7 +76,7 @@ args.device = "cuda"
 
 base_path = 'feishen29/IMAGDressing-v1'
 
-generator = torch.Generator(device=args.device).manual_seed(42)
+
 vae = AutoencoderKL.from_pretrained('./ckpt/sd-vae-ft-mse/').to(dtype=torch.float16, device=args.device)
 tokenizer = CLIPTokenizer.from_pretrained("./ckpt/tokenizer")
 text_encoder = CLIPTextModel.from_pretrained("./ckpt/text_encoder").to(dtype=torch.float16, device=args.device)
@@ -259,6 +259,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
     output = pipe(
         ref_image=vae_clothes,
         prompt=prompt,