Update generate_img.py

generate_img.py

@@ -6,7 +6,7 @@ from pipeline_stable_diffusion_controlnet_inpaint import *
 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import *
 import random
 #model1
-controlnet =  ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_openpose", torch_dtype=torch.float16,cache_dir='./models')
+controlnet =  ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_openpose", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     "./models/Any-inpainting", controlnet=controlnet, torch_dtype=torch.float16,cache_dir='./models'
 )
@@ -15,10 +15,10 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
 
 # model2
-controlnet1 =  ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_scribble",torch_dtype=torch.float16,cache_dir='./models')
+controlnet1 =  ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_scribble",torch_dtype=torch.float16)
 
 pipe1 = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-    "./models/Any-inpainting", controlnet=controlnet1, torch_dtype=torch.float16,cache_dir='./models'
+    "./models/Any-inpainting", controlnet=controlnet1, torch_dtype=torch.float16
 )
 
 pipe1.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)