gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-04 02:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: remove torch in main code (#439)

Browse Source 
* use np.ndarray in costeer model evaluators

* remove package

* fix test error

---------

Co-authored-by: Xu Yang <peteryang@vip.qq.com>
This commit is contained in:
Linlang
2024-10-21 15:56:04 +08:00
committed by GitHub
parent 2161956315
commit d3c59b2f33
7 changed files with 14 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/model_coder/model_execute_template_v1.txt
+1 -1
View File
@@ -37,7 +37,7 @@ if MODEL_TYPE == "Graph":
else:
out = m(data)
execution_model_output = out.cpu().detach()
execution_model_output = out.cpu().detach().numpy()
execution_feedback_str = f"Execution successful, output tensor shape: {execution_model_output.shape}"
pickle.dump(execution_model_output, open("execution_model_output.pkl", "wb"))