Update app.py
Browse files
app.py
CHANGED
@@ -68,12 +68,6 @@ pl_sd = torch.load(model_path_e, map_location="cuda")
|
|
68 |
sd = pl_sd["state_dict"]
|
69 |
model = instantiate_from_config(config.model)
|
70 |
m, u = model.load_state_dict(sd, strict=False)
|
71 |
-
if len(m) > 0 and verbose:
|
72 |
-
print("missing keys:")
|
73 |
-
print(m)
|
74 |
-
if len(u) > 0 and verbose:
|
75 |
-
print("unexpected keys:")
|
76 |
-
print(u)
|
77 |
|
78 |
model.half().to("cuda")
|
79 |
|
|
|
68 |
sd = pl_sd["state_dict"]
|
69 |
model = instantiate_from_config(config.model)
|
70 |
m, u = model.load_state_dict(sd, strict=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
|
72 |
model.half().to("cuda")
|
73 |
|