RamAnanth1 commited on
Commit
758cfc5
1 Parent(s): 928a499

Fix EOF literal

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -63,7 +63,7 @@ class Model:
63
  self.model_dir.mkdir(exist_ok=True, parents=True)
64
  device = 'cuda'
65
 
66
- config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml)
67
  config.model.params.cond_stage_config.params.device = device
68
 
69
  base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
 
63
  self.model_dir.mkdir(exist_ok=True, parents=True)
64
  device = 'cuda'
65
 
66
+ config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml")
67
  config.model.params.cond_stage_config.params.device = device
68
 
69
  base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"