Update model.py

model.py

@@ -85,7 +85,7 @@ class Model:
         ).to(device='cuda', dtype=precision)
 
         model_weight = load_file(gpt_ckpt)
-        gpt_model.load_state_dict(model_weight, strict=True)
+        # gpt_model.load_state_dict(model_weight, strict=True)
         gpt_model.eval()
         print("gpt model is loaded")
         return gpt_model