Update README.md

README.md

@@ -51,7 +51,13 @@ prompt = """Przykładowe zapytanie do modelu"""
 
 model_inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
 with torch.no_grad():
-  generated_ids = model.generate(**model_inputs, max_new_tokens=512, do_sample=True, penalty_alpha=0.6, top_k=5)
+  generated_ids = model.generate(
+    **model_inputs,
+    max_new_tokens=512,
+    do_sample=True,
+    penalty_alpha=0.6,
+    top_k=5
+  )
 
 output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 print(output)