-
Notifications
You must be signed in to change notification settings - Fork 49
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #38 from replicate/bump-ai-toolkit
Update AI toolkit
- Loading branch information
Showing
1 changed file
with
1 addition
and
1 deletion.
There are no files selected for viewing
Submodule ai-toolkit
updated
13 files
+2 −1 | .gitignore | |
+11 −1 | docker/Dockerfile | |
+53 −1 | extensions_built_in/sd_trainer/SDTrainer.py | |
+45 −6 | jobs/process/BaseSDTrainProcess.py | |
+2 −1 | requirements.txt | |
+22 −2 | toolkit/config_modules.py | |
+4 −2 | toolkit/custom_adapter.py | |
+1 −1 | toolkit/data_transfer_object/data_loader.py | |
+84 −0 | toolkit/logging.py | |
+492 −0 | toolkit/models/pixtral_vision.py | |
+58 −29 | toolkit/models/vd_adapter.py | |
+6 −0 | toolkit/sd_device_states_presets.py | |
+18 −7 | toolkit/stable_diffusion_model.py |