Spaces:

feishen29
/

IMAGDressing-v1

Running on Zero

feishen29 commited on Jul 24, 2024

Commit

06e57b1

verified ·

1 Parent(s): 7e882f5

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,7 +170,7 @@ noise_scheduler = DDIMScheduler(
 # noise_scheduler = UniPCMultistepScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
 control_net_openpose = ControlNetModel.from_pretrained(
-    "/home/sf/control_v11p_sd15_openpose",
     torch_dtype=torch.float16).to(device=args.device)
 # pipe = PipIpaControlNet(unet=unet, reference_unet=ref_unet, vae=vae, tokenizer=tokenizer,
 #                         text_encoder=text_encoder, image_encoder=image_encoder,

 # noise_scheduler = UniPCMultistepScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
 control_net_openpose = ControlNetModel.from_pretrained(
+    "./ckpt/control_v11p_sd15_openpose",
     torch_dtype=torch.float16).to(device=args.device)
 # pipe = PipIpaControlNet(unet=unet, reference_unet=ref_unet, vae=vae, tokenizer=tokenizer,
 #                         text_encoder=text_encoder, image_encoder=image_encoder,