Spaces:
Runtime error
Runtime error
fix errors
Browse files
scripts/rendertext_tool.py
CHANGED
@@ -15,9 +15,9 @@ from contextlib import nullcontext
|
|
15 |
|
16 |
def load_model_from_config(cfg, ckpt, verbose=False, not_use_ckpt=False):
|
17 |
|
18 |
-
if "model_ema.input_blocks10in_layers0weight" not in sd:
|
19 |
-
|
20 |
-
|
21 |
model = instantiate_from_config(cfg.model)
|
22 |
|
23 |
if ckpt.endswith("model_states.pt"):
|
@@ -31,7 +31,7 @@ def load_model_from_config(cfg, ckpt, verbose=False, not_use_ckpt=False):
|
|
31 |
nk = k[7:]
|
32 |
sd[nk] = sd[k]
|
33 |
del sd[k]
|
34 |
-
|
35 |
if not not_use_ckpt:
|
36 |
m, u = model.load_state_dict(sd, strict=False)
|
37 |
if len(m) > 0 and verbose:
|
|
|
15 |
|
16 |
def load_model_from_config(cfg, ckpt, verbose=False, not_use_ckpt=False):
|
17 |
|
18 |
+
# if "model_ema.input_blocks10in_layers0weight" not in sd:
|
19 |
+
# print("missing model_ema.input_blocks10in_layers0weight. set use_ema as False")
|
20 |
+
# cfg.model.params.use_ema = False
|
21 |
model = instantiate_from_config(cfg.model)
|
22 |
|
23 |
if ckpt.endswith("model_states.pt"):
|
|
|
31 |
nk = k[7:]
|
32 |
sd[nk] = sd[k]
|
33 |
del sd[k]
|
34 |
+
|
35 |
if not not_use_ckpt:
|
36 |
m, u = model.load_state_dict(sd, strict=False)
|
37 |
if len(m) > 0 and verbose:
|