wondervictor commited on
Commit
1be8d25
·
verified ·
1 Parent(s): 6c9f9c3

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -85,10 +85,10 @@ class Model:
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
 
 
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