Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,7 +71,7 @@ css = '''
|
|
| 71 |
<style>
|
| 72 |
'''
|
| 73 |
|
| 74 |
-
model_id = "
|
| 75 |
# model_path = "/home/qinghewang/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6"
|
| 76 |
pipe = StableDiffusionPipeline.from_pretrained(model_id) # , torch_dtype=torch.float16
|
| 77 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
|
|
|
| 71 |
<style>
|
| 72 |
'''
|
| 73 |
|
| 74 |
+
model_id = "Manojb/stable-diffusion-2-1-base"
|
| 75 |
# model_path = "/home/qinghewang/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6"
|
| 76 |
pipe = StableDiffusionPipeline.from_pretrained(model_id) # , torch_dtype=torch.float16
|
| 77 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|