p1atdev
/

dart-v2-base

Text Generation

text-generation-inference

Model card Files Files and versions

p1atdev commited on May 11, 2024

Commit

588265d

·

verified ·

1 Parent(s): a4bd284

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -73,13 +73,13 @@ from dartrs.dartrs import DartTokenizer
 from dartrs.utils import get_generation_config
 from dartrs.v2 import (
     compose_prompt,
-    MixtralModel,
     V2Model,
 )
 import time
 import os
 MODEL_NAME = "p1atdev/dart-v2-base"
-model = MixtralModel.from_pretrained(MODEL_NAME)
 tokenizer = DartTokenizer.from_pretrained(MODEL_NAME)
 config = get_generation_config(
     prompt=compose_prompt(

 from dartrs.utils import get_generation_config
 from dartrs.v2 import (
     compose_prompt,
+    MistralModel,
     V2Model,
 )
 import time
 import os
 MODEL_NAME = "p1atdev/dart-v2-base"
+model = MistralModel.from_pretrained(MODEL_NAME)
 tokenizer = DartTokenizer.from_pretrained(MODEL_NAME)
 config = get_generation_config(
     prompt=compose_prompt(