github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix import

pull/467/head
x54-729 2023-11-02 22:05:54 +08:00
parent 3418427083
commit 6b2ea75ca9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/transformers/convert2hf.py
View File

@ -6,8 +6,7 @@ import re
import tempfile
import torch
from modeling_internlm import InternLMConfig, InternLMForCausalLM
from tokenization_internlm import InternLMTokenizer
from internlm_model import InternLMConfig, InternLMForCausalLM, InternLMTokenizer
NUM_SHARDS = {
"7B": 1,