Vikhrmodels
/

Vikhr-7B-instruct_0.2

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

AlexWortega commited on Feb 12

Commit

8ce1638

•

1 Parent(s): 9b27654

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -10,7 +10,7 @@ from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 import torch
-MODEL_NAME = "Vikhrmodels/Vikhr-7B-instruct"
 DEFAULT_MESSAGE_TEMPLATE = "<s>{role}\n{content}</s>\n"
 DEFAULT_SYSTEM_PROMPT = "Ты — Вихрь, русскоязычный автоматический ассистент. Ты разговариваешь с людьми и помогаешь им."

 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 import torch
+MODEL_NAME = "Vikhrmodels/Vikhr-7B-instruct_0.2"
 DEFAULT_MESSAGE_TEMPLATE = "<s>{role}\n{content}</s>\n"
 DEFAULT_SYSTEM_PROMPT = "Ты — Вихрь, русскоязычный автоматический ассистент. Ты разговариваешь с людьми и помогаешь им."