Commit
•
0121054
1
Parent(s):
63e8784
Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ Refer https://github.com/davidkim205/komt
|
|
43 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
44 |
from transformers import TextStreamer, GenerationConfig
|
45 |
|
46 |
-
model_name='davidkim205/komt-llama2-
|
47 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|
48 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
49 |
streamer = TextStreamer(tokenizer)
|
|
|
43 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
44 |
from transformers import TextStreamer, GenerationConfig
|
45 |
|
46 |
+
model_name='davidkim205/komt-llama2-13b-v1'
|
47 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|
48 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
49 |
streamer = TextStreamer(tokenizer)
|