Jeongwon Choi
commited on
Commit
·
8a43dc6
1
Parent(s):
6723deb
Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ base_model: mistralai/Mistral-7B-Instruct-v0.2
|
|
| 8 |
pipeline_tag: text-generation
|
| 9 |
---
|
| 10 |
|
| 11 |
-
# **DataVortexM-7B-Instruct-v0.
|
| 12 |
<img src="./DataVortex.png" alt="DataVortex" style="height: 8em;">
|
| 13 |
|
| 14 |
## **License**
|
|
@@ -43,8 +43,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 43 |
|
| 44 |
device = "cuda"
|
| 45 |
|
| 46 |
-
model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexM-7B-Instruct-v0.
|
| 47 |
-
tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexM-7B-Instruct-v0.
|
| 48 |
|
| 49 |
messages = [
|
| 50 |
{ "role": "user", "content": "대한민국의 수도는 어디야?" }
|
|
|
|
| 8 |
pipeline_tag: text-generation
|
| 9 |
---
|
| 10 |
|
| 11 |
+
# **DataVortexM-7B-Instruct-v0.1**
|
| 12 |
<img src="./DataVortex.png" alt="DataVortex" style="height: 8em;">
|
| 13 |
|
| 14 |
## **License**
|
|
|
|
| 43 |
|
| 44 |
device = "cuda"
|
| 45 |
|
| 46 |
+
model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexM-7B-Instruct-v0.1", device_map=device)
|
| 47 |
+
tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexM-7B-Instruct-v0.1")
|
| 48 |
|
| 49 |
messages = [
|
| 50 |
{ "role": "user", "content": "대한민국의 수도는 어디야?" }
|