Update README.md

README.md

@@ -34,7 +34,7 @@ from datasets import load_dataset # Lets you load fine-tuning datasets
 model_id = "cvGod/DeepSeek-R1-Psychology-COT"
 model, tokenizer = FastLanguageModel.from_pretrained(
     model_name=model_id,
-    max_seq_length=2048,
+    max_seq_length=4096,
     dtype=None,
     load_in_4bit=True,
     
@@ -64,7 +64,7 @@ inputs = tokenizer([prompt_style.format(question, "")], return_tensors="pt").to(
 outputs = model.generate(
     input_ids=inputs.input_ids,          # Tokenized input IDs
     attention_mask=inputs.attention_mask, # Attention mask for padding handling
-    max_new_tokens=1024,                  # Maximum length for generated response
+    max_new_tokens=4096,                  # Maximum length for generated response
     use_cache=True,                        # Enable cache for efficient generation
 )