Spaces:
Running
on
Zero
Running
on
Zero
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")
|