Stanlito commited on
Commit
223289f
1 Parent(s): 3cfa49d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -36,7 +36,7 @@ It was trained on a Google Colab notebook with a T4 GPU and high RAM. It is main
36
  from transformers import AutoTokenizer
37
  import transformers
38
  import torch
39
- model = "mlabonne/llama-2-7b-stanlito"
40
  prompt = "What is a large language model?"
41
  tokenizer = AutoTokenizer.from_pretrained(model)
42
  pipeline = transformers.pipeline(
 
36
  from transformers import AutoTokenizer
37
  import transformers
38
  import torch
39
+ model = "Stanlito/llama-2-7b-stanlito"
40
  prompt = "What is a large language model?"
41
  tokenizer = AutoTokenizer.from_pretrained(model)
42
  pipeline = transformers.pipeline(