Spaces:
Runtime error
Runtime error
Update gradio_seesr_turbo.py
Browse files- 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=
|
| 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 |
|