Update README.md
Browse files
README.md
CHANGED
@@ -15,7 +15,7 @@ Note: Turkish was not in pretraining.
|
|
15 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
16 |
|
17 |
# Load model and tokenizer
|
18 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
19 |
model = AutoModelForSequenceClassification.from_pretrained('tum-nlp/neural-news-generator-llama-7b-fa')
|
20 |
|
21 |
# Create the pipeline for neural news generation and set the repetition penalty >1.1 to punish repetition.
|
|
|
15 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
16 |
|
17 |
# Load model and tokenizer
|
18 |
+
tokenizer = AutoTokenizer.from_pretrained(""mostafaamiri/persian_llama_7B_merged"")
|
19 |
model = AutoModelForSequenceClassification.from_pretrained('tum-nlp/neural-news-generator-llama-7b-fa')
|
20 |
|
21 |
# Create the pipeline for neural news generation and set the repetition penalty >1.1 to punish repetition.
|