Reduce model memory

.idea/workspace.xml

@@ -46,7 +46,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1677222707113</updated>
-      <workItem from="1677222708286" duration="4501000" />
+      <workItem from="1677222708286" duration="5087000" />
     </task>
     <servers />
   </component>

app.py

@@ -13,24 +13,7 @@ from huggingface_hub import hf_hub_url, cached_download
 
 
 REPO_ID = "lllyasviel/ControlNet"
-canny_checkpoint = "models/control_sd15_canny.pth"
 scribble_checkpoint = "models/control_sd15_scribble.pth"
-pose_checkpoint = "models/control_sd15_openpose.pth"
-
-canny_model = create_model('./models/cldm_v15.yaml').cpu()
-canny_model.load_state_dict(load_state_dict(cached_download(
-    hf_hub_url(REPO_ID, canny_checkpoint)
-), location='cpu'))
-canny_model = canny_model.cuda()
-ddim_sampler = DDIMSampler(canny_model)
-
-pose_model = create_model('./models/cldm_v15.yaml').cpu()
-pose_model.load_state_dict(load_state_dict(cached_download(
-    hf_hub_url(REPO_ID, pose_checkpoint)
-), location='cpu'))
-pose_model = pose_model.cuda()
-ddim_sampler_pose = DDIMSampler(pose_model)
-
 scribble_model = create_model('./models/cldm_v15.yaml').cpu()
 scribble_model.load_state_dict(load_state_dict(cached_download(
     hf_hub_url(REPO_ID, scribble_checkpoint)