Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -35,8 +35,7 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
35 |
),
|
36 |
tokenizer=tokenizer,
|
37 |
feature_extractor=feature_extractor,
|
38 |
-
torch_dtype=torch.
|
39 |
-
revision="fp16", # if your repo provides fp16 weights
|
40 |
use_safetensors=True
|
41 |
)
|
42 |
|
|
|
35 |
),
|
36 |
tokenizer=tokenizer,
|
37 |
feature_extractor=feature_extractor,
|
38 |
+
torch_dtype=torch.float32, # load weights in half-precision
|
|
|
39 |
use_safetensors=True
|
40 |
)
|
41 |
|