Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -120,7 +120,7 @@ pipe.watermark = NoWatermark()
|
|
120 |
refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
121 |
base_model_path, # we found use base_model_path instead of refiner_model_path may get a better performance
|
122 |
scheduler=scheduler,
|
123 |
-
|
124 |
use_safetensors=True
|
125 |
).to("cuda")
|
126 |
refiner.watermark = NoWatermark()
|
|
|
120 |
refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
121 |
base_model_path, # we found use base_model_path instead of refiner_model_path may get a better performance
|
122 |
scheduler=scheduler,
|
123 |
+
torch_dtype=torch.float16,
|
124 |
use_safetensors=True
|
125 |
).to("cuda")
|
126 |
refiner.watermark = NoWatermark()
|