Purple11 commited on
Commit
1840191
1 Parent(s): 8a3e99c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -259,7 +259,7 @@ def inference(input_prompt, input_category):
259
  seed_everything(opt.seed)
260
 
261
  tic = time.time()
262
- config = x.load(f"{opt.config}")
263
  model = load_model_from_config(config, f"{opt.sd_ckpt}")
264
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
265
  model = model.to(device)
 
259
  seed_everything(opt.seed)
260
 
261
  tic = time.time()
262
+ config = OmegaConf.load(f"{opt.config}")
263
  model = load_model_from_config(config, f"{opt.sd_ckpt}")
264
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
265
  model = model.to(device)