Commit
•
8510077
1
Parent(s):
38d686d
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ import transformers
|
|
| 22 |
from peft import PeftModel
|
| 23 |
|
| 24 |
# Model and tokenizer names
|
| 25 |
-
model_name = "lmsys/vicuna-7b-v1.5"
|
| 26 |
peft_model_id = "rudransh2004/FuturixAI-AmazonSentiment-LLAMA7B-LORA"
|
| 27 |
|
| 28 |
# Initialize the tokenizer and model
|
|
|
|
| 22 |
from peft import PeftModel
|
| 23 |
|
| 24 |
# Model and tokenizer names
|
| 25 |
+
model_name = "meta-llama/Llama-2-7b" #You can also use Lymsys LLAMA-2 finetuned Vicuna model alternatively "lmsys/vicuna-7b-v1.5"
|
| 26 |
peft_model_id = "rudransh2004/FuturixAI-AmazonSentiment-LLAMA7B-LORA"
|
| 27 |
|
| 28 |
# Initialize the tokenizer and model
|