Update gradio_app.py

gradio_app.py

@@ -120,23 +120,23 @@ cn_config = OmegaConf.load(cn_config_file)
 cn_model_config = cn_config.pop("control_stage_config", OmegaConf.create())
 
 print("before init") 
-model_list = []
-gpu_num = 1
-for gpu_id in range(gpu_num):
-    model = instantiate_from_config(model_config)
-    cn_model = instantiate_from_config(cn_model_config)
+# model_list = []
+# gpu_num = 1
+# for gpu_id in range(gpu_num):
+model = instantiate_from_config(model_config)
+cn_model = instantiate_from_config(cn_model_config)
 
-    # model = model.cuda(gpu_id)
-    assert os.path.exists(ckpt_path), "Error: checkpoint Not Found!"
-    model = load_model_checkpoint(model, ckpt_path)
-    model.eval()
+# model = model.cuda(gpu_id)
+assert os.path.exists(ckpt_path), "Error: checkpoint Not Found!"
+model = load_model_checkpoint(model, ckpt_path)
+model.eval()
 
-    cn_model.load_state_dict(load_state_dict(cn_ckpt_path, location='cpu'))  
-    cn_model.eval()
+cn_model.load_state_dict(load_state_dict(cn_ckpt_path, location='cpu'))  
+cn_model.eval()
 
-    model.control_model = cn_model
+model.control_model = cn_model
     
-    model_list.append(model)
+    # model_list.append(model)
 
 save_fps = 8
 print("resolution:", resolution)
@@ -161,8 +161,8 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
     if steps > 60:
         steps = 60
 
-    global model_list 
-    model = model_list[gpu_id]
+    global model 
+    # model = model_list[gpu_id]
     model = model.cuda()
     
     batch_size=1