Update README.md
Browse files
README.md
CHANGED
@@ -49,13 +49,13 @@ bnb_config = BitsAndBytesConfig(
|
|
49 |
)
|
50 |
|
51 |
model = AutoModelForCausalLM.from_pretrained(
|
52 |
-
"msaad02/
|
53 |
quantization_config=bnb_config,
|
54 |
device_map={"": 0},
|
55 |
trust_remote_code=True
|
56 |
)
|
57 |
|
58 |
-
tokenizer = AutoTokenizer.from_pretrained("msaad02/
|
59 |
|
60 |
# Use a pipeline as a high-level helper
|
61 |
from transformers import pipeline
|
|
|
49 |
)
|
50 |
|
51 |
model = AutoModelForCausalLM.from_pretrained(
|
52 |
+
"msaad02/llama2_7b_brockportgpt",
|
53 |
quantization_config=bnb_config,
|
54 |
device_map={"": 0},
|
55 |
trust_remote_code=True
|
56 |
)
|
57 |
|
58 |
+
tokenizer = AutoTokenizer.from_pretrained("msaad02/llama2_7b_brockportgpt")
|
59 |
|
60 |
# Use a pipeline as a high-level helper
|
61 |
from transformers import pipeline
|