Update README.md

README.md

@@ -43,19 +43,18 @@ with torch.no_grad():
     token_ids = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
     output_ids = model.generate(
         token_ids.to(model.device),
-        max_new_tokens=256,
+        max_new_tokens=1024,
         do_sample=True,
         temperature=0.3,
         top_p=0.95,
         top_k=50,
-        repetition_penalty=1.1,
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
     )
 output = tokenizer.decode(output_ids.tolist()[0][token_ids.size(1) :], skip_special_tokens=False)
 print(output)
 """
-
+旅行先の都市が急速に発展している可能性があります。高層ビルは一般的に、経済の成長、人口の増加、不動産開発の活発化など、都市の発展に関連する要因と結びついています。
 """
 ```
 
@@ -67,7 +66,7 @@ print(output)
 
 ### Hyperparameter
 ```
-num_train_epochs: 5
+num_train_epochs: ~5~ 4
 per_device_train_batch_size: 2
 per_device_eval_batch_size: 2
 gradient_accumulation_steps: 128