Refactoring docker images, updating CUDA to 12.6 and pytorch 2.6
This commit is contained in:
1 parent
73d9d9450b
commit
284c100860
11 files changed
+123
-143
No files matched your search
+7
-1
@@ -29,6 +29,12 @@ while [ "$#" -gt 0 ]; do
|
||||
DOCKER_TAG="$2"
|
||||
shift
|
||||
;;
|
||||
--github-repository)
|
||||
if [ -n "${GITHUB_REPOSITORY}" ] && ! [[ $GITHUB_REPOSITORY =~ ^--.* ]]; then
|
||||
GITHUB_REPOSITORY="$2"
|
||||
shift
|
||||
fi
|
||||
;;
|
||||
*)
|
||||
# Allow to pass arbitrary arguments to docker as well to be flexible:
|
||||
ADDITIONAL_ARGS+=( $1 )
|
||||
@@ -65,4 +71,4 @@ docker run \
|
||||
--name alltalk \
|
||||
"${DOCKER_ARGS[@]}" \
|
||||
"${ADDITIONAL_ARGS[@]}" \
|
||||
alltalk_beta:${DOCKER_TAG} &> /dev/stdout
|
||||
${GITHUB_REPOSITORY}alltalk_beta:${DOCKER_TAG} &> /dev/stdout
|
||||
Reference in new issue
Block a user