Update README.md

README.md

@@ -8,21 +8,7 @@ datasets:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 
-with ('generation_config.json').open('w') as fp:
-    json.dump({
-        "pad_token_id": 0,
-        "bos_token_id": 1,
-        "eos_token_id": 2,
-        "temperature": 0.3,
-        "top_p": 0.9,
-        "top_k": 50,
-        "do_sample": True,
-        "max_new_tokens": 1536,
-        "repetition_penalty": 1.1,
-        "no_repeat_ngram_size": 15,
-    }, fp, indent=4)
-
-MODEL_NAME = "Vikhrmodels/Vikhr_instruct"
+MODEL_NAME = "Vikhrmodels/Vikhr-7B-instruct"
 TEMPLATE = "<s>{role}\n{content}</s>\n"
 SYSTEM_PROMPT = "Ты – полезный помощник по имени Вихрь. Ты разговариваешь с людьми и помогаешь им."
 
@@ -31,7 +17,7 @@ model.to('cuda')
 model.eval()
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=False)
-generation_config = GenerationConfig.from_pretrained("generation_config.json")
+generation_config = GenerationConfig.from_pretrained(MODEL_NAME)
 
 class Conversation:
     def __init__(