cbensimon HF staff commited on
Commit
17cb182
·
1 Parent(s): 9199c87
Files changed (1) hide show
  1. app.py +1 -1
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(f"{pretrained_model_path}/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)
 
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)