|
|
@@ -337,7 +337,7 @@ ARG API_SERVER_PORT=8080
|
|
|
ARG LLAMA_CHECKPOINT_PATH="checkpoints/s2-pro"
|
|
|
ARG DECODER_CHECKPOINT_PATH="checkpoints/s2-pro/codec.pth"
|
|
|
ARG DECODER_CONFIG_NAME="modded_dac_vq"
|
|
|
-
|
|
|
+ARG FISH_API_SERVER_ARGS="{}"
|
|
|
# Expose port
|
|
|
EXPOSE ${API_SERVER_PORT}
|
|
|
|
|
|
@@ -347,6 +347,7 @@ ENV API_SERVER_PORT=${API_SERVER_PORT}
|
|
|
ENV LLAMA_CHECKPOINT_PATH=${LLAMA_CHECKPOINT_PATH}
|
|
|
ENV DECODER_CHECKPOINT_PATH=${DECODER_CHECKPOINT_PATH}
|
|
|
ENV DECODER_CONFIG_NAME=${DECODER_CONFIG_NAME}
|
|
|
+ENV FISH_API_SERVER_ARGS=${FISH_API_SERVER_ARGS}
|
|
|
|
|
|
# Create server entrypoint
|
|
|
RUN printf '%s\n' \
|