michaelfeil
commited on
Commit
·
09e91a7
1
Parent(s):
cc8bde5
Upload mosaicml/mpt-7b-chat ctranslate fp16 weights
Browse files
README.md
CHANGED
@@ -20,14 +20,14 @@ Speedup inference while reducing memory by 2x-4x using int8 inference in C++ on
|
|
20 |
|
21 |
quantized version of [mosaicml/mpt-7b-chat](https://huggingface.co/mosaicml/mpt-7b-chat)
|
22 |
```bash
|
23 |
-
pip install hf-hub-ctranslate2>=2.0.8
|
24 |
```
|
25 |
-
Converted on 2023-05-
|
26 |
```
|
27 |
-
ct2-transformers-converter --model mosaicml/mpt-7b-chat --output_dir /home/michael/tmp-ct2fast-mpt-7b-chat --force --copy_files
|
28 |
```
|
29 |
|
30 |
-
Checkpoint compatible to [ctranslate2>=3.
|
31 |
- `compute_type=int8_float16` for `device="cuda"`
|
32 |
- `compute_type=int8` for `device="cpu"`
|
33 |
|
@@ -46,7 +46,8 @@ model = GeneratorCT2fromHfHub(
|
|
46 |
)
|
47 |
outputs = model.generate(
|
48 |
text=["How do you call a fast Flan-ingo?", "User: How are you doing? Bot:"],
|
49 |
-
max_length=64
|
|
|
50 |
)
|
51 |
print(outputs)
|
52 |
```
|
|
|
20 |
|
21 |
quantized version of [mosaicml/mpt-7b-chat](https://huggingface.co/mosaicml/mpt-7b-chat)
|
22 |
```bash
|
23 |
+
pip install hf-hub-ctranslate2>=2.0.8 ctranslate2>=3.14.0
|
24 |
```
|
25 |
+
Converted on 2023-05-31 using
|
26 |
```
|
27 |
+
ct2-transformers-converter --model mosaicml/mpt-7b-chat --output_dir /home/michael/tmp-ct2fast-mpt-7b-chat --force --copy_files tokenizer.json README.md tokenizer_config.json generation_config.json special_tokens_map.json .gitattributes --quantization float16 --trust_remote_code
|
28 |
```
|
29 |
|
30 |
+
Checkpoint compatible to [ctranslate2>=3.14.0](https://github.com/OpenNMT/CTranslate2) and [hf-hub-ctranslate2>=2.0.8](https://github.com/michaelfeil/hf-hub-ctranslate2)
|
31 |
- `compute_type=int8_float16` for `device="cuda"`
|
32 |
- `compute_type=int8` for `device="cpu"`
|
33 |
|
|
|
46 |
)
|
47 |
outputs = model.generate(
|
48 |
text=["How do you call a fast Flan-ingo?", "User: How are you doing? Bot:"],
|
49 |
+
max_length=64,
|
50 |
+
include_prompt_in_result=False
|
51 |
)
|
52 |
print(outputs)
|
53 |
```
|