Spaces:
Runtime error
Runtime error
Commit
Β·
021be0e
1
Parent(s):
55d0d55
Upload folder using huggingface_hub
Browse files- aagenerator.py +1 -1
aagenerator.py
CHANGED
@@ -597,7 +597,7 @@ class Generator:
|
|
597 |
# args.pretrained_model_name_or_path, subfolder="text_encoder", revision=args.revision
|
598 |
# )
|
599 |
self.vae = AutoencoderKL.from_pretrained(
|
600 |
-
"/
|
601 |
)
|
602 |
|
603 |
vae_path = (
|
|
|
597 |
# args.pretrained_model_name_or_path, subfolder="text_encoder", revision=args.revision
|
598 |
# )
|
599 |
self.vae = AutoencoderKL.from_pretrained(
|
600 |
+
"stabilityai/sd-vae-ft-mse"
|
601 |
)
|
602 |
|
603 |
vae_path = (
|