diff --git a/model_examples/Pi-0/README.md b/model_examples/Pi-0/README.md index 153eb563c229e7995ba4963897cf47ef4284cbe6..d67a255273836922c7bd2ec6ef7734168f1e5267 100644 --- a/model_examples/Pi-0/README.md +++ b/model_examples/Pi-0/README.md @@ -118,7 +118,7 @@ 2. 下载[pi-0预训练权重](https://huggingface.co/lerobot/pi0),将权重路径记作 pi0_weights ## 快速开始 -本任务主要提供**单机8卡**的训练脚本。在训练前,需要在torch_run.sh文件中修改对应路径信息。 +本任务主要提供**单机8卡**的训练脚本。 ### 开始训练 - 单机8卡性能 @@ -153,5 +153,5 @@ Q: 在无网络或者有防火墙的网络下,模型无法自动下载paligemm A: 可自行下载[paligemma权重](https://huggingface.co/google/paligemma-3b-pt-224),将权重路径记作 paligemma_weights。再执行以下命令,使用脚本将本地权重路径替换进模型代码: ``` - bash test/paligemma_path_mod.sh {paligemma_weights} + bash test/paligemma_weights_mod.sh {paligemma_weights} ``` \ No newline at end of file diff --git a/model_examples/Senna/Senna.patch b/model_examples/Senna/Senna.patch index 29ba90cad23cf11cacf216f2f7e71e0aac97c293..672d48716333d256b51a641600f9b3579090d9d4 100644 --- a/model_examples/Senna/Senna.patch +++ b/model_examples/Senna/Senna.patch @@ -60,17 +60,18 @@ index 8fa159c..c2db4bd 100644 class LlavaConfig(LlamaConfig): model_type = "llava_llama" diff --git a/requirements.txt b/requirements.txt -index f1df450..57617c5 100644 +index f1df450..1cd5768 100644 --- a/requirements.txt +++ b/requirements.txt -@@ -1,6 +1,5 @@ +@@ -1,7 +1,6 @@ absl-py==2.1.0 accelerate==0.29.2 -aidisdk==0.18.0 -aiohttp==3.9.5 -+aiohttp==3.12.14 ++aiohttp aiohttp-cors==0.7.0 aiomultiprocess==0.9.1 + aiosignal==1.3.1 @@ -72,7 +71,6 @@ msgpack==1.0.8 multidict==6.0.5 multiprocess==0.70.15 @@ -79,6 +80,15 @@ index f1df450..57617c5 100644 networkx==3.3 ninja==1.11.1.1 nltk==3.8.1 +@@ -88,7 +86,7 @@ pillow==10.3.0 + platformdirs==4.2.1 + prometheus_client==0.20.0 + proto-plus==1.23.0 +-protobuf==4.25.3 ++protobuf + psutil==5.9.8 + py-cpuinfo==9.0.0 + py-spy==0.3.14 @@ -108,7 +106,6 @@ python-dateutil==2.9.0.post0 python-dotenv==1.0.1 pytz==2024.1