yyk19 commited on
Commit
86f09db
·
1 Parent(s): e7a5f93

fix errors

Browse files
Files changed (1) hide show
  1. scripts/rendertext_tool.py +4 -4
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
- 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,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: