Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ snapshot_download(
|
|
| 77 |
pretrained_model_path = 'preset/models/stable-diffusion-2-1-base'
|
| 78 |
seesr_model_path = 'preset/models/seesr'
|
| 79 |
|
| 80 |
-
scheduler =
|
| 81 |
text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
|
| 82 |
tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
|
| 83 |
vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
|
|
|
|
| 77 |
pretrained_model_path = 'preset/models/stable-diffusion-2-1-base'
|
| 78 |
seesr_model_path = 'preset/models/seesr'
|
| 79 |
|
| 80 |
+
scheduler = DDIMScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
|
| 81 |
text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
|
| 82 |
tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
|
| 83 |
vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
|