Commit History

Enhance run_training function in app.py to ensure child Python processes are unbuffered for real-time logging. Update _prepare_script to include unbuffered flag and modify UI launch to support ssr_mode, improving compatibility with different Gradio versions.
3382b9b

yeq6x commited on

Refactor UI in app.py to implement a tabbed layout for the Trainer and Prompt Generator sections, enhancing organization and user experience. The update includes restructured input fields and improved accessibility for image uploads and prompt generation functionalities.
c29e4f9

yeq6x commited on

Enhance _list_checkpoints function in app.py to skip files that are zero-length, too new, or unreadable. This improvement ensures more robust file handling by filtering out potentially in-flight files, thereby reducing errors during checkpoint listing.
542e0cd

yeq6x commited on

Add zero-length file check in _list_checkpoints function of app.py to skip empty files during checkpoint listing. This enhancement improves the robustness of file handling by preventing errors related to empty files.
7c0568c

yeq6x commited on

Refactor error handling in run_training function of app.py to remove unnecessary None yields. This change streamlines the output by ensuring consistent return values when errors occur, enhancing clarity in the training process.
f87e901

yeq6x commited on

Update prefix/suffix naming in _prepare_script of app.py to ensure consistency with target parameters. This change improves clarity in metadata creation by correctly labeling the flags for target prefix and suffix.
c347e01

yeq6x commited on

Refactor UI layout in app.py by reorganizing the prefix/suffix input fields into accordions for improved accessibility and visual coherence. This update enhances the user experience during image uploads by maintaining a consistent structure across control sections.
50994c2

yeq6x commited on

Refactor UI layout in app.py to improve organization of target image input fields. Adjust the arrangement of the prefix and suffix textboxes for better accessibility and visual coherence, enhancing the overall user experience during image uploads.
e70e3a5

yeq6x commited on

Refactor UI in app.py to enhance layout and styling of sections, improving organization and accessibility. Adjust padding and margins for better visual consistency, and update accordion titles for clarity on prefix/suffix functionality in image uploads.
144203d

yeq6x commited on

Refactor UI layout in app.py by consolidating input fields into accordions for better organization and accessibility. This update enhances the user experience for managing target and control images, improving the overall interface for image uploads and training configurations.
c396bb4

yeq6x commited on

Refactor UI layout in app.py to improve organization and accessibility of input fields by grouping related elements and adding custom CSS styling. This enhances the overall user experience for image uploads and training parameter configurations.
9da1071

yeq6x commited on

Refactor UI layout in app.py to enhance user experience by incorporating an accordion for control_0 image previews. This change improves organization and accessibility of image upload and preview functionalities.
b423631

yeq6x commited on

Add hyperparameter overriding functionality in _prepare_script of app.py to allow safer and more flexible adjustments of learning rate, network dimension, seed, max training epochs, and save frequency. This enhances configurability for training scripts.
c5b4ecf

yeq6x commited on

Refactor app.py to update prefix/suffix naming conventions for metadata creation and enhance UI with new training hyperparameter inputs. Modify train_QIE.sh to utilize dynamic hyperparameter values for training execution, improving configurability and user experience.
904c1ea

yeq6x commited on

Add hyperparameter override options in _prepare_script function of app.py to allow dynamic adjustment of learning rate, network dimension, and seed during script preparation. Update UI elements for clarity in target image handling.
b29f092

yeq6x commited on

Implement target prefix/suffix functionality in app.py and create_image_caption_json.py for improved image processing. Update UI to reflect changes, enhancing user experience with clearer labeling and file handling. Refactor related functions to accommodate new naming conventions.
8ad19ed

yeq6x commited on

Add checkpoint listing functionality in app.py to track and display model checkpoints during training. Update run_training to yield checkpoint information and enhance Gradio UI with checkpoint file outputs for improved user experience.
78b8ad3

yeq6x commited on

Refactor app.py to set output_dir_base to the unique dataset directory and enhance Gradio UI launch by allowing access to specific runtime directories, improving file serving capabilities.
480fc74

yeq6x commited on

Update app.py to remove outdated startup instructions from the UI, enhance train_QIE.sh to dynamically set cache_directory, and modify requirements.txt to specify a custom index URL for torchvision installation.
bf20d03

yeq6x commited on

Refactor logging in run_training function of app.py to accumulate log messages in a buffer before yielding, improving output consistency and readability during training execution.
0538a11

yeq6x commited on

Add dependency installation for musubi-tuner in app.py during startup. Implement best-effort upgrade for pip, setuptools, and wheel, and handle optional Torch extras for editable installation. Fallback to plain install if editable fails, ensuring smoother setup process.
c888ffa

yeq6x commited on

Fix indentation in app.py for control_upload_sets enumeration in run_training function, improving code readability and structure.
4f98208

yeq6x commited on

Enhance image upload handling in app.py by implementing PNG conversion during file saving and adding prefix/suffix options for metadata creation. Update UI to include fields for main and control image prefixes/suffixes, improving user experience and flexibility in image processing.
8ece4a2

yeq6x commited on

Add normalization function for filename keys in app.py and update GPU decorator for run_training function
be11537

yeq6x commited on

Refactor UI in app.py to clearly separate mandatory control_0 image uploads from optional control images. Control_0 uploads are now displayed outside of the accordion for improved user guidance, while optional controls are organized within a closed accordion by default.
3915e6f

yeq6x commited on

Update run_training function in app.py to enforce mandatory control_0 image uploads, improving error handling and user guidance. Modify UI to reflect that control_0 images are required, enhancing clarity for users during image uploads.
dc4e3c7

yeq6x commited on

Refactor run_training function in app.py to return tuples for error and log messages, enhancing clarity in output handling. Update UI to remove user input for models_root, output_dir_base, and dataset_config, which are now resolved at runtime. Implement logic to locate the latest LoRA file for download after training completion.
81ee3f1

yeq6x commited on

Enhance app.py by adding an override option for run name in the _prepare_script function and updating the run_training function to use an auto-generated dataset directory name. Modify UI to reflect changes from dataset_name to output_name for better clarity in user input.
6d7d2b1

yeq6x commited on

Refactor run_training function in app.py to simplify parameters and enhance control image handling. Update UI for image uploads and ensure at least one control folder is provided for training. Adjust data root resolution to auto-decide based on user input.
0a92c44

yeq6x commited on

Reorder startup functions in app.py to ensure musubi-tuner is cloned before downloading models. This change maintains the intended execution flow during application startup.
3357f66

yeq6x commited on

Enhance app.py with new functions for handling data uploads and directory management. Implement data root resolution and control set preparation for training. Update main execution flow to accommodate new upload handling and directory structure.
bb0426b

yeq6x commited on

Refactor app.py to support runtime-resolved paths for musubi-tuner and models directories. Implement fallback mechanisms for directory creation and cloning operations. Update UI components to reflect new runtime paths.
3ae9658

yeq6x commited on

Update default paths in app.py and download_qwen_image_models.py to use /data and /auto directories. Adjust metadata generation script to reflect new working directory structure.
630a29d

yeq6x commited on

Enhance training functionality with ZeroGPU support and UI adjustments. Added options to override max epochs and save frequency, and implemented GPU request handling for Spaces compatibility.
177760c

yeq6x commited on

Add musubi-tuner integration: clone repository at startup and set default directory and repo URL
6160205

yeq6x commited on

Add import for spaces module in app.py
71e1e3a

yeq6x commited on

Implement initial project structure and setup
701afe3

yeq6x commited on