Update finetune_sample/README.md

finetune_sample/README.md

@@ -8,12 +8,13 @@ python3 -m venv gptq_finetune
 source gptq_finetune/bin/activate
 
 
-pip install transformers==4.32.1
+pip install transformers==4.34.1
 pip install datasets
 pip install peft==0.5.0
 pip install trl
 pip install auto-gptq
 pip install optimum
+pip install torch==2.0.1
 
 # finetune qlora
 python3 finetune.py
@@ -30,15 +31,17 @@ ValueError: Target module QuantLinear() is not supported. Currently, only `torch
 It's because peft old version.  
 
 I don't know if it's required, but the version of my running environment.  
-* auto-gptq                0.4.1+cu117
-* trl                      0.7.1
-* optimum                  1.12.1.dev0
-* datasets                 2.14.4
+* auto-gptq                0.4.2
+* trl                      0.7.2
+* optimum                  1.13.2
+* datasets                 2.14.6
 
 The official documentation says to install from source, but sometimes that causes errors.  
 If you can't get it to work, it might be better to wait until the stable version comes out.  
 Good luck!  
 
+If you encounter "RuntimeError: Unrecognized tensor type ID: AutocastCUDA", chech your torch version.  
+
 
 - finetune.py gptq finetune sample file.
 - jawiki3.csv sample data.(Japanese)