subfolder
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ scheduler = UniPCMultistepScheduler.from_pretrained(pretrained_model_path, subfo
|
|
60 |
text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
|
61 |
tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
|
62 |
vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
|
63 |
-
feature_extractor = CLIPImageProcessor.from_pretrained(
|
64 |
unet = UNet2DConditionModel.from_pretrained(ckpt_path, subfolder="unet")
|
65 |
controlnet = ControlNetModel.from_pretrained(ckpt_path, subfolder="controlnet")
|
66 |
vae.requires_grad_(False)
|
|
|
60 |
text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
|
61 |
tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
|
62 |
vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
|
63 |
+
feature_extractor = CLIPImageProcessor.from_pretrained(pretrained_model_path, subfolder="feature_extractor")
|
64 |
unet = UNet2DConditionModel.from_pretrained(ckpt_path, subfolder="unet")
|
65 |
controlnet = ControlNetModel.from_pretrained(ckpt_path, subfolder="controlnet")
|
66 |
vae.requires_grad_(False)
|