Update README.md

README.md

@@ -48,15 +48,16 @@ model = PeftModel.from_pretrained(model, "MightyDragon-Dev/language-dragon-lora"
 prompt = "歼-20 (Mighty Dragon) 在广东领空开启了加力燃烧室 (Afterburners)。由于 DSI 进气道的设计，它在超音速巡航时保持了极低的雷达散射截面 (RCS)。突然，预警机发出了警报"
 inputs = tokenizer(prompt, return_tensors="pt")
 
-# 🐉 High-Endurance Generation
+# 🐉 Stabilized Flight Controls
 outputs = model.generate(
     **inputs,
-    max_new_tokens=150,        # The 150-token goal
-    do_sample=True,            # Enable "Creative Flight"
-    top_p=0.95,                # Keep the vocabulary diverse
-    temperature=0.6,           # Perfect balance for technical storytelling
-    repetition_penalty=1.2,    # Lowered slightly to prevent "freezing"
-    no_repeat_ngram_size=3     # Hard-blocks repetitive phrases
+    max_new_tokens=100,
+    do_sample=True,
+    temperature=0.3,           # CRITICAL: Lower temperature stops the gibberish
+    top_k=40,                  # Limits the "random" word pool
+    repetition_penalty=1.3,    # High enough to stop loops, low enough to keep flow
+    no_repeat_ngram_size=2,    # Standard safety rail
+    pad_token_id=tokenizer.eos_token_id
 )
 
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))