Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,7 @@ tags:
|
|
| 9 |
- llama
|
| 10 |
- trl
|
| 11 |
base_model: unsloth/llama-3-8b-bnb-4bit
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# Uploaded model
|
|
@@ -36,4 +37,4 @@ inputs = tokenizer("[POST START]: New Architecture that replaces the MLP by usin
|
|
| 36 |
from transformers import TextStreamer
|
| 37 |
text_streamer = TextStreamer(tokenizer)
|
| 38 |
_ = model.generate(**inputs, streamer = text_streamer, max_new_tokens = 1000, repetition_penalty=1.4)
|
| 39 |
-
```
|
|
|
|
| 9 |
- llama
|
| 10 |
- trl
|
| 11 |
base_model: unsloth/llama-3-8b-bnb-4bit
|
| 12 |
+
pipeline_tag: text-generation
|
| 13 |
---
|
| 14 |
|
| 15 |
# Uploaded model
|
|
|
|
| 37 |
from transformers import TextStreamer
|
| 38 |
text_streamer = TextStreamer(tokenizer)
|
| 39 |
_ = model.generate(**inputs, streamer = text_streamer, max_new_tokens = 1000, repetition_penalty=1.4)
|
| 40 |
+
```
|