Spaces:
Sleeping
Sleeping
build demo retry
Browse files- 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 |
-
#
|
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")
|