v1.0.1

README.md

@@ -33,7 +33,7 @@ pip install -r requirements.txt # requirements.txt
 ```
 ### Dockerfile（方法二）
 ```
-cd MiniCPM/docker
+cd gemma_pytorch/docker
 docker build --no-cache -t gemma:latest .
 docker run --shm-size=16G --name gemma -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../gemma_pytorch:/home/gemma_pytorch -it gemma bash
 # 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。
@@ -53,7 +53,7 @@ triton:2.1.0
 
 2、其它非特殊库参照requirements.txt安装
 ```
-pip install -r finetune/requirements.txt # finetune/requirements.txt
+pip install -r finetune/requirements.txt # requirements.txt
 ```
 
 ## 数据集

infer.sh

 #/bin/bash
+export PYTHONPATH=$PYTHONPATH:`pwd`/gemma
 python scripts/run.py --ckpt="gemma-2b-pytorch/gemma-2b-it.ckpt" --variant=2b --prompt="The meaning of life is" --device=cuda