update code

app.py

@@ -1016,6 +1016,7 @@ def launch_gradio_app():
 
 if __name__ == '__main__':
     import torch.multiprocessing as mp
+    import transformers
     mp.set_start_method('spawn', force=True)
     # logging.info("Environment Variables: %s" % os.environ)
     # logging.info("Installing CUDA extensions...")
@@ -1040,8 +1041,14 @@ if __name__ == '__main__':
         controlnet_path, torch_dtype=torch.float16
     )
     sd_path =  './pretrained_model/sd21'
+    text_encoder = transformers.CLIPTextModel.from_pretrained(
+            sd_path,
+            subfolder="text_encoder",
+            num_hidden_layers=12 - (2 - 1),
+            torch_dtype=torch.float16
+        )    
     pipeline_sd = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-        sd_path, torch_dtype=torch.float16,
+        sd_path, torch_dtype=torch.float16, text_encoder=text_encoder,
         use_safetensors=True, controlnet=controlnet, variant="fp16"
     ).to(device)
     demo_cam = False