Update model.py

model.py

@@ -90,6 +90,7 @@ 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
+        device = 'cuda' 
         # if current_base != base_model:
         #     ckpt = os.path.join("models", base_model)
         #     pl_sd = torch.load(ckpt, map_location="cpu")