RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update run_modal_docker.py to use /app/ai-toolkit paths instead of /root/ai-toolkit

This commit is contained in:
hndrr 2025-07-15 14:46:06 +09:00
parent bc4fbb3fd6
commit 12163a1922
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
run_modal_docker.py
View File

@ -2,7 +2,7 @@
ostris/ai-toolkit on https://modal.com
Run training with the following command:
modal run run_modal.py --config-file-list-str=/root/ai-toolkit/config/whatever_you_want.yml
modal run run_modal.py --config-file-list-str=/app/ai-toolkit/config/whatever_you_want.yml
'''
@ -14,7 +14,7 @@ from dotenv import load_dotenv
# Load the .env file if it exists
load_dotenv()
sys.path.insert(0, "/root/ai-toolkit")
sys.path.insert(0, "/app/ai-toolkit")
# must come before ANY torch or fastai imports
# import toolkit.cuda_malloc
@ -26,10 +26,10 @@ os.environ['DISABLE_TELEMETRY'] = 'YES'
model_volume = modal.Volume.from_name("flux-lora-models", create_if_missing=True)
# modal_output, due to "cannot mount volume on non-empty path" requirement
MOUNT_DIR = "/root/ai-toolkit/modal_output" # modal_output, due to "cannot mount volume on non-empty path" requirement
MOUNT_DIR = "/app/ai-toolkit/modal_output" # modal_output, due to "cannot mount volume on non-empty path" requirement
# define modal app
image = modal.Image.from_dockerfile("docker/Dockerfile").add_local_dir("config", "/root/ai-toolkit/config").add_local_dir("input/images", '/root/ai-toolkit/input/images')
image = modal.Image.from_dockerfile("docker/Dockerfile").add_local_dir("config", "/app/ai-toolkit/config").add_local_dir("input/images", '/app/ai-toolkit/input/images')
# mount for the entire ai-toolkit directory
# example: "/Users/username/ai-toolkit" is the local directory, "/root/ai-toolkit" is the remote directory