Update model.py

model.py

@@ -62,10 +62,8 @@ class Model:
     def download_models(self) -> None:
         self.model_dir.mkdir(exist_ok=True, parents=True)
         device = 'cuda'
-        
-        config_file = config_path+'test_sketch.yaml'
-        subprocess.run(shlex.split(f'wget {config_file} -O config_sketch.yaml'))    
-        config = OmegaConf.load("config_sketch.yaml")
+    
+        config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml)
         config.model.params.cond_stage_config.params.device = device
 
         base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"