Skip to content

H
HN-CodeGen
Commit c0fd4e00 by zhengzifu
Options

Update Python version to 3.12 and optimize matrix handling in… 

Update Python version to 3.12 and optimize matrix handling in generate_wt_group.py by transposing the matrix before shape extraction.
parent 05df0620
Showing with 2 additions and 1 deletions
hllm/optimized/generate_wt_group.py
...@@ -113,6 +113,7 @@ def run(name: str, config: CFG): ...@@ -113,6 +113,7 @@ def run(name: str, config: CFG):
with open(weights_file, "rb") as f: with open(weights_file, "rb") as f:
print(f"Loading {weights_file_name}") print(f"Loading {weights_file_name}")
matrixs = pickle.load(f) matrixs = pickle.load(f)
matrixs = np.transpose(matrixs, (1, 0, 2))
VN, L, H = matrixs.shape VN, L, H = matrixs.shape
with ProcessPoolExecutor(max_workers=config.num_workers) as executor: with ProcessPoolExecutor(max_workers=config.num_workers) as executor:
futures = [ futures = [
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment