apolinario commited on
Commit
8c61b5b
·
1 Parent(s): 5b21366

File name match

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def load_model_from_config(config, ckpt, verbose=False):
42
  return model
43
 
44
  config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml") # TODO: Optionally download from same location as ckpt and chnage this logic
45
- model = load_model_from_config(config, f"latent_diffusion_txt2img_f8_large.ckpt") # TODO: check path
46
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
47
  model = model.to(device)
48
 
 
42
  return model
43
 
44
  config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml") # TODO: Optionally download from same location as ckpt and chnage this logic
45
+ model = load_model_from_config(config, f"txt2img-f8-large.ckpt") # TODO: check path
46
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
47
  model = model.to(device)
48