Spaces:
Running
on
Zero
Running
on
Zero
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 |
|