Update README.md
Browse files
README.md
CHANGED
@@ -76,7 +76,7 @@ Use the code below to get started with the model.
|
|
76 |
import torch
|
77 |
from transformers import pipeline
|
78 |
|
79 |
-
model_id = "kyutai/helium-1-preview"
|
80 |
|
81 |
pipe = pipeline(
|
82 |
"text-generation",
|
|
|
76 |
import torch
|
77 |
from transformers import pipeline
|
78 |
|
79 |
+
model_id = "kyutai/helium-1-preview-2b"
|
80 |
|
81 |
pipe = pipeline(
|
82 |
"text-generation",
|