Update README.md

README.md

@@ -48,11 +48,12 @@ messages = [
 ]
 chat_tokenized = tokenizer.apply_chat_template(messages, tokenize=False)
 
-model = AutoModelForCausalLM.from_pretrained("szymonrucinski/krakowiak-v2-7b")
+model = AutoModelForCausalLM.from_pretrained("szymonrucinski/Krakowiak-7B-v2")
 model.config.pad_token_id = model.config.eos_token_id
 
-tokenizer = AutoTokenizer.from_pretrained("szymonrucinski/krakowiak-v2-7b",add_eos_token=True)
+tokenizer = AutoTokenizer.from_pretrained("szymonrucinski/Krakowiak-7B-v2",add_eos_token=True)
 tokenizer.pad_token = tokenizer.eos_token
+model_inputs = tokenizer(chat_tokenized, return_tensors='pt').to(torch_device)
 
 beam_outputs = model.generate(
     **model_inputs,
@@ -62,7 +63,6 @@ beam_outputs = model.generate(
     num_return_sequences=1,
     early_stopping=True
 )
-model_inputs = tokenizer(chat_tokenized, return_tensors='pt').to(torch_device)
 print(tokenizer.decode(beam_outputs[0], skip_special_tokens=True))
 
 ```