Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ pip install torch transformers==4.40.0 accelerate
|
|
80 |
import transformers
|
81 |
import torch
|
82 |
|
83 |
-
model_id = "MLP-KTLim/Bllossom"
|
84 |
|
85 |
pipeline = transformers.pipeline(
|
86 |
"text-generation",
|
@@ -132,7 +132,7 @@ import os
|
|
132 |
import torch
|
133 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
134 |
|
135 |
-
model_id = 'MLP-KTLim/Bllossom'
|
136 |
|
137 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
138 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
80 |
import transformers
|
81 |
import torch
|
82 |
|
83 |
+
model_id = "MLP-KTLim/llama3-Bllossom"
|
84 |
|
85 |
pipeline = transformers.pipeline(
|
86 |
"text-generation",
|
|
|
132 |
import torch
|
133 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
134 |
|
135 |
+
model_id = 'MLP-KTLim/llama3-Bllossom'
|
136 |
|
137 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
138 |
model = AutoModelForCausalLM.from_pretrained(
|