Update README.md

README.md

@@ -174,7 +174,7 @@ stopping_criteria_list = StoppingCriteriaList([stopping_criteria])
 
 #EXAMPLE #1
 input_text = """James Buchanan es el único presidente que nunca se casó.”"""
-targets_traslate= "en"
+targets_traslate= "en" # response english.
 
 txt=f"""<bos><start_of_turn>system
 You are a helpful AI assistant.
@@ -193,13 +193,13 @@ lista de codigos linguisticos disponibles: ["es", "en", "fr", "de"].<end_of_turn
 inputs = tokenizer.encode(txt,
                           return_tensors="pt",
                           add_special_tokens=False).to("cuda:0")
-max_new_tokens=1000
+max_new_tokens=200
 generation_config = GenerationConfig(
               max_new_tokens=max_new_tokens,
-              temperature=0.55,
+              temperature=0.32,
               #top_p=0.9,
-              #top_k=len_tokens,
-              repetition_penalty=1.1, 
+              top_k=45,
+              repetition_penalty=1., 
               do_sample=True,
           )
 outputs = model.generate(generation_config=generation_config,