Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,12 +14,7 @@ if torch.cuda.is_available():
|
|
14 |
else:
|
15 |
torch_dtype = torch.float32
|
16 |
|
17 |
-
pipe = StableDiffusionXLPipeline.
|
18 |
-
ckpt_path,
|
19 |
-
use_safetensors=True,
|
20 |
-
torch_dtype=torch.float16,
|
21 |
-
)
|
22 |
-
|
23 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
24 |
pipe.scheduler.register_to_config(
|
25 |
prediction_type="v_prediction",
|
|
|
14 |
else:
|
15 |
torch_dtype = torch.float32
|
16 |
|
17 |
+
pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
|
|
|
|
|
|
|
|
|
|
18 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
19 |
pipe.scheduler.register_to_config(
|
20 |
prediction_type="v_prediction",
|