cmykk commited on
Commit
f27bb99
1 Parent(s): 68dcd67

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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('bigscience/bloomz')
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.