Hugging Face's logo

Spaces:
loooooong
/
StableGarment
Runtime error

App Files Community
loooooong commited on
Commit
10dfd9a
•
1 Parent(s): 0da8a10

update model path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ base_model_path = "SG161222/Realistic_Vision_V4.0_noVAE"
31
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(dtype=torch_dtype,device=device)
32
  scheduler = UniPCMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5",subfolder="scheduler")
33
 
34
- pretrained_garment_encoder_path = "StableGarment_text2img"
35
  garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
36
  garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
37
 
 
31
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(dtype=torch_dtype,device=device)
32
  scheduler = UniPCMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5",subfolder="scheduler")
33
 
34
+ pretrained_garment_encoder_path = "loooooong/StableGarment_text2img"
35
  garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
36
  garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
37