Update model.py

model.py

@@ -90,15 +90,15 @@ class Model:
     @torch.inference_mode()
     def process_sketch(self, input_img, type_in, color_back, prompt, neg_prompt, fix_sample, scale, con_strength, base_model):    
         global current_base
-        if current_base != base_model:
-            ckpt = os.path.join("models", base_model)
-            pl_sd = torch.load(ckpt, map_location="cpu")
-            if "state_dict" in pl_sd:
-                sd = pl_sd["state_dict"]
-            else:
-                sd = pl_sd
-            model.load_state_dict(sd, strict=False) #load_model_from_config(config, os.path.join("models", base_model)).to(device)
-            current_base = base_model
+        # if current_base != base_model:
+        #     ckpt = os.path.join("models", base_model)
+        #     pl_sd = torch.load(ckpt, map_location="cpu")
+        #     if "state_dict" in pl_sd:
+        #         sd = pl_sd["state_dict"]
+        #     else:
+        #         sd = pl_sd
+        #     model.load_state_dict(sd, strict=False) #load_model_from_config(config, os.path.join("models", base_model)).to(device)
+        #     current_base = base_model
         con_strength = int((1-con_strength)*50)
         if fix_sample == 'True':
             seed_everything(42)