deepseekmirror/DeepSeek-V3
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-V3.git synced 2025-02-23 14:18:57 -05:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

small patch

Browse Source 
removed unnecessary variable
This commit is contained in:
CodingParadigm1 2025-02-02 06:14:17 -07:00
parent 35244be39f
commit 07de76f5ee
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
inference/convert.py
View File

@ -72,7 +72,6 @@ def main(hf_ckpt_path, save_path, n_experts, mp):
torch.set_num_threads(8) torch.set_num_threads(8)
n_local_experts = n_experts // mp n_local_experts = n_experts // mp
state_dicts = [{} for _ in range(mp)] state_dicts = [{} for _ in range(mp)]
tasks = []
tensor_dir = glob(os.path.join(hf_ckpt_path, "*.safetensors")) tensor_dir = glob(os.path.join(hf_ckpt_path, "*.safetensors"))
token_dir = glob(os.path.join(hf_ckpt_path, "*token*")) token_dir = glob(os.path.join(hf_ckpt_path, "*token*"))
for file_path in tqdm(tensor_dir): for file_path in tqdm(tensor_dir):