修改了生文时的bug

build_space/Dockerfile.jupyterlab_ubuntu

@@ -92,15 +92,21 @@ RUN if [ -n "$TENSORFLOW_VERSION" ]; then \
 
 # ----- paddlepaddle install -----
 RUN if [ -n "$PADDLEPADDLE_VERSION" ] && [ -n "$PADDLE_URL" ]; then \
-    pip install paddlepaddle-gpu==$PADDLEPADDLE_VERSION -f $PADDLE_URL  \
+    pip install paddlepaddle-gpu==$PADDLEPADDLE_VERSION -f $PADDLE_URL -i $PADDLE_URL \
     && rm -r /root/.cache/pip; \
 fi
 
-RUN if [ -n "$PADDLENLP_VERSION" ] && [ -z "$PADDLE_URL" ]; then \
-    pip install paddlenlp==$PADDLENLP_VERSION -i https://pypi.tuna.tsinghua.edu.cn/simple \
+RUN if [ -n "$PADDLEPADDLE_VERSION" ] && [ -z "$PADDLE_URL" ]; then \
+    pip install paddlepaddle-gpu==$PADDLEPADDLE_VERSION -i https://pypi.tuna.tsinghua.edu.cn/simple \
     && rm -r /root/.cache/pip; \
 fi
 
+RUN if [ -n "$PADDLENLP_VERSION" ] ; then \
+    pip install paddlenlp==$PADDLENLP_VERSION ppdiffusers huggingface_hub --no-cache-dir  -i https://pypi.tuna.tsinghua.edu.cn/simple && \
+    pip install --upgrade ppdiffusers --no-deps  && rm -r /root/.cache/pip; \
+fi
+
+
 COPY ./python-requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/python-requirements.txt
 

build_space/python-requirements.txt

@@ -5,4 +5,3 @@ matplotlib
 transformers
 git-lfs
 accelerate
-diffusers
\ No newline at end of file

script/2_text_generate_test.sh

 #!/bin/bash
-ARGET_DIR=gpu-base-image-test/paddletest
+TARGET_DIR=gpu-base-image-test/paddletest
 docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace $1 python text.py