修改README

README.md

@@ -24,40 +24,6 @@ Vicuna：即LLM，用<video>来代表视频内容，将视觉特征Z嵌入到tex
     <img src="./doc/VTimeLLM.PNG"/>
 </div>
 
-## 代码改动说明
-deepspeed0.14.2只能使用transformers4.31.0，此版本不支持k100ai使用bf16/tf32精度，修改transformers代码，避免精度错误：
-
-ps：仓库中是改动后的代码，不需再次修改
-```
-pip show pip  #查看依赖库安装地址site-packages
-
-1、site-packages/transformers/utils/import_utils.py，修改def is_torch_bf16_gpu_available():
-...
-#TODO:if torch.cuda.is_available() and torch.version.cuda is not None:
-if torch.cuda.is_available():
-    if torch.cuda.get_device_properties(torch.cuda.current_device()).major < 8:
-        return False
-    # if int(torch.version.cuda.split(".")[0]) < 11:
-       # return False
-    if not hasattr(torch.cuda.amp, "autocast"):
-        return False
-else:
-    return False
-
-2、site-packages/transformers/utils/import_utils.py，修改def is_torch_tf32_available():
-...
-#TODO:if not torch.cuda.is_available() or torch.version.cuda is None:
-if not torch.cuda.is_available():
-    return False
-if torch.cuda.get_device_properties(torch.cuda.current_device()).major < 8:
-    return False
-# if int(torch.version.cuda.split(".")[0]) < 11:
-   #  return False
-if version.parse(version.parse(torch.__version__).base_version) < version.parse("1.7"):
-    return False
-return True
-```
-
 ## 环境配置
 ### Docker（方法一）
 ```
@@ -110,6 +76,41 @@ export HF_ENDPOINT=https://hf-mirror.com
 apt update
 apt install gcc libaio-dev
 ```
+
+## 代码改动说明
+deepspeed0.14.2只能使用transformers4.31.0，此版本不支持k100ai使用bf16/tf32精度，修改transformers代码，避免精度错误：
+
+ps：仓库中是改动后的代码，不需再次修改
+```
+pip show pip  #查看依赖库安装地址site-packages
+
+1、site-packages/transformers/utils/import_utils.py，修改def is_torch_bf16_gpu_available():
+...
+#TODO:if torch.cuda.is_available() and torch.version.cuda is not None:
+if torch.cuda.is_available():
+    if torch.cuda.get_device_properties(torch.cuda.current_device()).major < 8:
+        return False
+    # if int(torch.version.cuda.split(".")[0]) < 11:
+       # return False
+    if not hasattr(torch.cuda.amp, "autocast"):
+        return False
+else:
+    return False
+
+2、site-packages/transformers/utils/import_utils.py，修改def is_torch_tf32_available():
+...
+#TODO:if not torch.cuda.is_available() or torch.version.cuda is None:
+if not torch.cuda.is_available():
+    return False
+if torch.cuda.get_device_properties(torch.cuda.current_device()).major < 8:
+    return False
+# if int(torch.version.cuda.split(".")[0]) < 11:
+   #  return False
+if version.parse(version.parse(torch.__version__).base_version) < version.parse("1.7"):
+    return False
+return True
+```
+
 ## 数据集
 ### 训练数据集
 VTimeLLM可基于Vicuna v1.5训练英文版本、基于ChatGLM3-6b训练中文版本，训练某个版本时只下载对应的数据集（data不同）即可。