JeongwonChoi
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": "λνλ―Όκ΅μ μλλ μ΄λμΌ?" }
|