Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ pip install torch transformers==4.40.0 accelerate
|
|
39 |
import transformers
|
40 |
import torch
|
41 |
|
42 |
-
model_id = "
|
43 |
|
44 |
pipeline = transformers.pipeline(
|
45 |
"text-generation",
|
@@ -90,7 +90,7 @@ import os
|
|
90 |
import torch
|
91 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
92 |
|
93 |
-
model_id = '
|
94 |
|
95 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
96 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
39 |
import transformers
|
40 |
import torch
|
41 |
|
42 |
+
model_id = "kfkas/Hansung-Bllossom-8B"
|
43 |
|
44 |
pipeline = transformers.pipeline(
|
45 |
"text-generation",
|
|
|
90 |
import torch
|
91 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
92 |
|
93 |
+
model_id = 'kfkas/Hansung-Bllossom-8B'
|
94 |
|
95 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
96 |
model = AutoModelForCausalLM.from_pretrained(
|