Spaces:
Sleeping
Sleeping
Commit
·
ae35b0a
1
Parent(s):
a12d1d7
Update app.py
Browse files
app.py
CHANGED
@@ -35,8 +35,10 @@ if mps_available:
|
|
35 |
torch_device = "cpu"
|
36 |
torch_dtype = torch.float32
|
37 |
|
|
|
38 |
t2i_pipe = StableDiffusionPipeline.from_single_file(
|
39 |
"https://huggingface.co/wanghuging/skin_demo/blob/main/skin_demo.safetensors",
|
|
|
40 |
safety_checker = None,
|
41 |
requires_safety_checker = False
|
42 |
)
|
|
|
35 |
torch_device = "cpu"
|
36 |
torch_dtype = torch.float32
|
37 |
|
38 |
+
scheduler = EulerDiscreteScheduler.from_pretrained(repo_id, subfolder="scheduler")
|
39 |
t2i_pipe = StableDiffusionPipeline.from_single_file(
|
40 |
"https://huggingface.co/wanghuging/skin_demo/blob/main/skin_demo.safetensors",
|
41 |
+
scheduler=scheduler,
|
42 |
safety_checker = None,
|
43 |
requires_safety_checker = False
|
44 |
)
|