Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Akegarasu committed Aug 30, 2024
1 parent 274f8c9 commit e319e7c
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 5 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ __pycache__
output/*
!output/.keep

assets/config.json

py310
python
git
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ RUN git clone --recurse-submodules https://github.com/Akegarasu/lora-scripts
WORKDIR /app/lora-scripts
RUN pip install xformers==0.0.27.post2 --no-deps && pip install -r requirements.txt

WORKDIR /app/lora-scripts/sd-scripts
WORKDIR /app/lora-scripts/scripts
RUN pip install -r requirements.txt

WORKDIR /app/lora-scripts
Expand Down
2 changes: 1 addition & 1 deletion mikazuki/app/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ async def run_script(request: Request, background_tasks: BackgroundTasks):
value = f'"{v}"'
result.append(value)
script_args = " ".join(result)
script_path = Path(os.getcwd()) / "sd-scripts" / script_name
script_path = Path(os.getcwd()) / "scripts" / script_name
cmd = f"{launch_utils.python_bin} {script_path} {script_args}"
background_tasks.add_task(launch_utils.run, cmd)
return APIResponseSuccess()
Expand Down
2 changes: 1 addition & 1 deletion svd_merge.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ if ($new_conv_rank) {
}

# run svd_merge
accelerate launch --num_cpu_threads_per_process=8 "./sd-scripts/networks/svd_merge_lora.py" `
accelerate launch --num_cpu_threads_per_process=8 "./scripts/networks/svd_merge_lora.py" `
--save_precision=$save_precision `
--precision=$precision `
--new_rank=$new_rank `
Expand Down
4 changes: 2 additions & 2 deletions train.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
},
"outputs": [],
"source": [
"!accelerate launch --num_cpu_threads_per_process=8 \"./sd-scripts/train_network.py\" \\\n",
"!accelerate launch --num_cpu_threads_per_process=8 \"./scripts/train_network.py\" \\\n",
" --enable_bucket \\\n",
" --pretrained_model_name_or_path=$pretrained_model \\\n",
" --train_data_dir=$train_data_dir \\\n",
Expand Down Expand Up @@ -96,4 +96,4 @@
},
"nbformat": 4,
"nbformat_minor": 2
}
}

0 comments on commit e319e7c

Please sign in to comment.