alexnasa commited on
Commit
dda6c02
·
verified ·
1 Parent(s): d080da1

Update gradio_seesr_turbo.py

Browse files
Files changed (1) hide show
  1. gradio_seesr_turbo.py +2 -2
gradio_seesr_turbo.py CHANGED
@@ -55,7 +55,7 @@ scheduler = DDPMScheduler.from_pretrained(pretrained_model_path, subfolder="sche
55
  text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
56
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
57
  vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
58
- feature_extractor = CLIPImageProcessor.from_pretrained(f"{pretrained_model_path}/feature_extractor")
59
  unet = UNet2DConditionModel.from_pretrained_orig(seesr_model_path, subfolder="unet")
60
  controlnet = ControlNetModel.from_pretrained(seesr_model_path, subfolder="controlnet")
61
 
@@ -73,7 +73,7 @@ else:
73
 
74
  # Get the validation pipeline
75
  validation_pipeline = StableDiffusionControlNetPipeline(
76
- vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, feature_extractor=feature_extractor,
77
  unet=unet, controlnet=controlnet, scheduler=scheduler, safety_checker=None, requires_safety_checker=False,
78
  )
79
 
 
55
  text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
56
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
57
  vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
58
+ # feature_extractor = CLIPImageProcessor.from_pretrained(f"{pretrained_model_path}/feature_extractor")
59
  unet = UNet2DConditionModel.from_pretrained_orig(seesr_model_path, subfolder="unet")
60
  controlnet = ControlNetModel.from_pretrained(seesr_model_path, subfolder="controlnet")
61
 
 
73
 
74
  # Get the validation pipeline
75
  validation_pipeline = StableDiffusionControlNetPipeline(
76
+ vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, feature_extractor=None,
77
  unet=unet, controlnet=controlnet, scheduler=scheduler, safety_checker=None, requires_safety_checker=False,
78
  )
79