Update README.md
Browse files
README.md
CHANGED
@@ -47,9 +47,9 @@ import torch
|
|
47 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
48 |
|
49 |
# Load the HelpingAI2-4x6B model
|
50 |
-
model = AutoModelForCausalLM.from_pretrained("
|
51 |
# Load the tokenizer
|
52 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
53 |
|
54 |
|
55 |
# Define the chat input
|
|
|
47 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
48 |
|
49 |
# Load the HelpingAI2-4x6B model
|
50 |
+
model = AutoModelForCausalLM.from_pretrained("Abhaykoul/HelpingAI2-4x6B ", trust_remote_code=True)
|
51 |
# Load the tokenizer
|
52 |
+
tokenizer = AutoTokenizer.from_pretrained("Abhaykoul/HelpingAI2-4x6B ", trust_remote_code=True)
|
53 |
|
54 |
|
55 |
# Define the chat input
|