01-ai
/

Yi-6B

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

14 contributors

History: 15 commits

cArlIcon's picture

make flash-attn optional

37a9590 8 months ago

.gitattributes

1.52 kB

initial commit 8 months ago
LICENSE

17.2 kB

update LICENSE and README 8 months ago
README.md

5.28 kB

Update README.md (#4) 8 months ago
config.json

717 Bytes

update config 8 months ago
configuration_yi.py

5.47 kB

add Yi-6B model and tokenizer 8 months ago
generation_config.json

132 Bytes

add Yi-6B model and tokenizer 8 months ago
md5

184 Bytes

add Yi-6B model and tokenizer 8 months ago
model-00001-of-00002.safetensors

9.94 GB
LFS

add safetensors format 8 months ago
model-00002-of-00002.safetensors

2.18 GB
LFS

add safetensors format 8 months ago
model.safetensors.index.json

22.9 kB

add safetensors format 8 months ago
modeling_yi.py

40.5 kB

make flash-attn optional 8 months ago
pytorch_model-00001-of-00002.bin
Detected Pickle imports (3)
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch.BFloat16Storage"
What is a pickle import?
9.94 GB
LFS

add Yi-6B model and tokenizer 8 months ago
pytorch_model-00002-of-00002.bin
Detected Pickle imports (3)
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch.BFloat16Storage"
What is a pickle import?
2.18 GB
LFS

add Yi-6B model and tokenizer 8 months ago
pytorch_model.bin.index.json

22.9 kB

add Yi-6B model and tokenizer 8 months ago
tokenization_yi.py

8.96 kB

add Yi-6B model and tokenizer 8 months ago
tokenizer.model

1.03 MB
LFS

add Yi-6B model and tokenizer 8 months ago
tokenizer_config.json

199 Bytes

add Yi-6B model and tokenizer 8 months ago