Harley-ml
/

MiniMD-28M

text-generation-inference

Model card Files Files and versions

Harley-ml commited on 8 days ago

Commit

62892fe

·

verified ·

1 Parent(s): fedddce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -236,7 +236,7 @@ from typing import Optional
 from transformers import AutoTokenizer, MistralForCausalLM
 # --------- CONFIG ----------
-MODEL_DIR = "Harley-ml/Mini-MD"
 TOKENIZER_DIR = MODEL_DIR
 DEFAULT_MAX_NEW_TOKENS = 640
 DEFAULT_TEMPERATURE = 0.9

 from transformers import AutoTokenizer, MistralForCausalLM
 # --------- CONFIG ----------
+MODEL_DIR = "Harley-ml/MiniMD-28M"
 TOKENIZER_DIR = MODEL_DIR
 DEFAULT_MAX_NEW_TOKENS = 640
 DEFAULT_TEMPERATURE = 0.9