add debug log

This commit is contained in:
strint 2025-10-16 21:42:48 +08:00
parent 2b222962c3
commit c1eac555c0

View File

@ -291,6 +291,7 @@ class BaseModel(torch.nn.Module):
return out
def load_model_weights(self, sd, unet_prefix=""):
import pdb; pdb.set_trace()
to_load = {}
keys = list(sd.keys())
for k in keys: