Erwann Millon commited on
Commit
0be9cd5
1 Parent(s): 28c5269

change default path

Browse files
Files changed (1) hide show
  1. loaders.py +2 -2
loaders.py CHANGED
@@ -17,10 +17,10 @@ def load_config(config_path, display=False):
17
 
18
 
19
  def load_default(device):
20
- conf_path = "./celeba_vqgan/vqgan_only.yaml"
21
  config = load_config(conf_path, display=False)
22
  model = taming.models.vqgan.VQModel(**config.model.params)
23
- sd = torch.load("./celeba_vqgan/vqgan_only.pt", map_location=device)
24
  model.load_state_dict(sd, strict=True)
25
  model.to(device)
26
  del sd
 
17
 
18
 
19
  def load_default(device):
20
+ conf_path = "./model_checkpoints/vqgan_only.yaml"
21
  config = load_config(conf_path, display=False)
22
  model = taming.models.vqgan.VQModel(**config.model.params)
23
+ sd = torch.load("./model_checkpoints/vqgan_only.pt", map_location=device)
24
  model.load_state_dict(sd, strict=True)
25
  model.to(device)
26
  del sd