Spaces:
Runtime error
Runtime error
del variant
Browse files
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,
|
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 |
|