Spaces:
Running
on
Zero
Running
on
Zero
wondervictor
commited on
Update model.py
Browse files
model.py
CHANGED
@@ -85,10 +85,10 @@ class Model:
|
|
85 |
).to(device='cpu', dtype=precision)
|
86 |
|
87 |
model_weight = load_file(gpt_ckpt)
|
88 |
-
|
89 |
gpt_model.load_state_dict(model_weight, strict=True)
|
90 |
gpt_model.eval()
|
91 |
-
|
92 |
print("gpt model is loaded")
|
93 |
return gpt_model
|
94 |
|
|
|
85 |
).to(device='cpu', dtype=precision)
|
86 |
|
87 |
model_weight = load_file(gpt_ckpt)
|
88 |
+
print("prev:", model_weight['adapter.model.embeddings.patch_embeddings.projection.weight'])
|
89 |
gpt_model.load_state_dict(model_weight, strict=True)
|
90 |
gpt_model.eval()
|
91 |
+
print("loaded:", gpt_model.adapter.model.embeddings.patch_embeddings.projection.weight)
|
92 |
print("gpt model is loaded")
|
93 |
return gpt_model
|
94 |
|