Update README.md

README.md

@@ -105,8 +105,7 @@ model = AutoModelForCausalLM.from_pretrained(
     model_name,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.bfloat32,
-    device_map="auto",
-    use_cache=False
+    device_map="auto"
 )
 
 # Example agricultural query
@@ -131,7 +130,8 @@ with torch.no_grad():
         top_k=50,
         top_p=0.95,
         temperature=0.6,
-        eos_token_id=tokenizer.eos_token_id
+        eos_token_id=tokenizer.eos_token_id,
+        use_cache=False
     )
 
 print(tokenizer.decode(output[0], skip_special_tokens=True))