Spaces:
Runtime error
Runtime error
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 |
|