Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
PATH = '
|
2 |
from PIL import Image
|
3 |
import torch
|
4 |
import torch.distributed as dist
|
@@ -133,7 +133,7 @@ if __name__ == "__main__":
|
|
133 |
|
134 |
|
135 |
dpm = DPMSolverMultistepScheduler.from_pretrained(PATH, subfolder="scheduler")
|
136 |
-
pipe = StableDiffusionPipeline.
|
137 |
|
138 |
demo.queue(concurrency_count=2,
|
139 |
).launch()
|
|
|
1 |
+
PATH = 'harpomaxx/deeplili' #stable diffusion 1.5
|
2 |
from PIL import Image
|
3 |
import torch
|
4 |
import torch.distributed as dist
|
|
|
133 |
|
134 |
|
135 |
dpm = DPMSolverMultistepScheduler.from_pretrained(PATH, subfolder="scheduler")
|
136 |
+
pipe = StableDiffusionPipeline.from_pretrained(PATH,torch_dtype=dtype, scheduler=dpm).to(device)
|
137 |
|
138 |
demo.queue(concurrency_count=2,
|
139 |
).launch()
|