loooooong commited on
Commit
1a06c05
1 Parent(s): 7d115cb

del variant

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ pretrained_garment_encoder_path = "loooooong/StableGarment_text2img"
36
  garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
37
  garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
38
 
39
- pipeline_t2i = StableGarmentPipeline.from_pretrained(base_model_path, vae=vae, torch_dtype=torch_dtype, variant="fp16").to(device=device)
40
  # pipeline = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", vae=vae, torch_dtype=torch_dtype, variant="fp16").to(device=device)
41
  pipeline_t2i.scheduler = scheduler
42
 
 
36
  garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
37
  garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
38
 
39
+ pipeline_t2i = StableGarmentPipeline.from_pretrained(base_model_path, vae=vae, torch_dtype=torch_dtype,).to(device=device) # variant="fp16"
40
  # pipeline = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", vae=vae, torch_dtype=torch_dtype, variant="fp16").to(device=device)
41
  pipeline_t2i.scheduler = scheduler
42