Update inference_manager.py
Browse files- inference_manager.py +1 -1
inference_manager.py
CHANGED
@@ -161,7 +161,7 @@ class InferenceManager:
|
|
161 |
#unet=unet,
|
162 |
torch_dtype=torch.bfloat16,
|
163 |
use_safetensors=True,
|
164 |
-
variant="fp16",
|
165 |
custom_pipeline = "lpw_stable_diffusion_xl",
|
166 |
)
|
167 |
|
|
|
161 |
#unet=unet,
|
162 |
torch_dtype=torch.bfloat16,
|
163 |
use_safetensors=True,
|
164 |
+
#variant="fp16",
|
165 |
custom_pipeline = "lpw_stable_diffusion_xl",
|
166 |
)
|
167 |
|