Spaces:

rashmi
/

h2oai-predict-llm

Sleeping

rashmi commited on Nov 9, 2023

Commit

4c68a95

•

1 Parent(s): f4d8307

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -184,7 +184,7 @@ def do_inference(full_text):
     # model = CustomModel()
     state = torch.load(model_paths[0], map_location=torch.device("cpu"))
-    model.load_state_dict(state["model"])  # ,strict=False)
     model.eval()
     model.to(device)

     # model = CustomModel()
     state = torch.load(model_paths[0], map_location=torch.device("cpu"))
+    model.load_state_dict(state["model"] ,strict=False)
     model.eval()
     model.to(device)