Spaces:
Running
on
Zero
Running
on
Zero
chore: use bf16
Browse files- diffusion.py +1 -1
diffusion.py
CHANGED
@@ -80,7 +80,7 @@ class ImageGenerator:
|
|
80 |
def __init__(self, model_name: str = "cagliostrolab/animagine-xl-3.1"):
|
81 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
82 |
model_name,
|
83 |
-
torch_dtype=torch.
|
84 |
use_safetensors=True,
|
85 |
add_watermarker=False,
|
86 |
custom_pipeline="lpw_stable_diffusion_xl",
|
|
|
80 |
def __init__(self, model_name: str = "cagliostrolab/animagine-xl-3.1"):
|
81 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
82 |
model_name,
|
83 |
+
torch_dtype=torch.bfloat16,
|
84 |
use_safetensors=True,
|
85 |
add_watermarker=False,
|
86 |
custom_pipeline="lpw_stable_diffusion_xl",
|