Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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)
|