Update README.md
Browse files
README.md
CHANGED
@@ -101,7 +101,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
101 |
|
102 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
103 |
|
104 |
-
model_name = "openGPT-X/Teuken-7B-instruct-v0.4"
|
105 |
model = AutoModelForCausalLM.from_pretrained(
|
106 |
model_name,
|
107 |
trust_remote_code=True,
|
|
|
101 |
|
102 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
103 |
|
104 |
+
model_name = "openGPT-X/Teuken-7B-instruct-research-v0.4"
|
105 |
model = AutoModelForCausalLM.from_pretrained(
|
106 |
model_name,
|
107 |
trust_remote_code=True,
|