Update src/pipeline.py
Browse files- src/pipeline.py +1 -2
src/pipeline.py
CHANGED
|
@@ -28,8 +28,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
|
|
| 28 |
if not pipeline:
|
| 29 |
pipeline = StableDiffusionXLPipeline.from_pretrained(
|
| 30 |
"stablediffusionapi/newdream-sdxl-20",
|
| 31 |
-
|
| 32 |
-
quantization_config=nf4_config,
|
| 33 |
).to("cuda")
|
| 34 |
|
| 35 |
# Register optimizations for performance
|
|
|
|
| 28 |
if not pipeline:
|
| 29 |
pipeline = StableDiffusionXLPipeline.from_pretrained(
|
| 30 |
"stablediffusionapi/newdream-sdxl-20",
|
| 31 |
+
torch_dtype=torch.float16,
|
|
|
|
| 32 |
).to("cuda")
|
| 33 |
|
| 34 |
# Register optimizations for performance
|