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 |
|