Update model.py

model.py

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