yyk19 commited on
Commit
13649aa
1 Parent(s): b9d42ae

build demo retry

Browse files
Files changed (1) hide show
  1. transfer.py +1 -1
transfer.py CHANGED
@@ -3,7 +3,7 @@ from scripts.rendertext_tool import Render_Text, load_model_from_config
3
  import torch
4
 
5
  # cfg = OmegaConf.load("other_configs/config_ema.yaml")
6
- # # model = load_model_from_config(cfg, "model_states.pt", verbose=True)
7
  # model = load_model_from_config(cfg, "mp_rank_00_model_states.pt", verbose=True)
8
 
9
  cfg = OmegaConf.load("other_configs/config_ema_unlock.yaml")
 
3
  import torch
4
 
5
  # cfg = OmegaConf.load("other_configs/config_ema.yaml")
6
+ # model = load_model_from_config(cfg, "model_states.pt", verbose=True)
7
  # model = load_model_from_config(cfg, "mp_rank_00_model_states.pt", verbose=True)
8
 
9
  cfg = OmegaConf.load("other_configs/config_ema_unlock.yaml")