初始化项目,由ModelHub XC社区提供模型
Model: SpectraSuite/TriLM_1.1B_Unpacked Source: Original Platform
This commit is contained in:
21
README.md
Normal file
21
README.md
Normal file
@@ -0,0 +1,21 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
---
|
||||
|
||||
# TriLM 1.1B Unpacked
|
||||
|
||||
TriLM (ternary model), unpacked to FP16 format - compatible with FP16 GEMMs. After unpacking, TriLM has the same architecture as LLaMa.
|
||||
|
||||
```python
|
||||
import transformers as tf, torch
|
||||
model_name = "SpectraSuite/TriLM_1.1B_Unpacked"
|
||||
|
||||
# Please adjust the temperature, repetition penalty, top_k, top_p and other sampling parameters according to your needs.
|
||||
pipeline = tf.pipeline("text-generation", model=model_id, model_kwargs={"torch_dtype": torch.float16}, device_map="auto")
|
||||
|
||||
# These are base (pretrained) LLMs that are not instruction and chat tuned. You may need to adjust your prompt accordingly.
|
||||
pipeline("Once upon a time")
|
||||
```
|
||||
|
||||
* License: Apache 2.0
|
||||
* We will use our GitHub repo for communication (including HF repo related queries). Feel free to open an issue here https://github.com/NolanoOrg/SpectraSuite
|
||||
Reference in New Issue
Block a user