This commit introduces a series of refactoring and enhancements across multiple modules, improving code readability, consistency, and adding new functionalities.
Highlights:
1. `tests/test_hf_diffusers_nodes.py`:
- Removed unnecessary whitespace.
- Improved variable assignment consistency (e.g., `ret=ret[0]` to `ret = ret[0]`).
- Added a new test case `test_create_unet2d_model` for Unet2d model creation.
2. `server_endpoints.py`:
- Cleaned up commented-out code and added more descriptive comments.
- Refactored the construction of `real_git_path` for better readability and future maintenance.
- Implemented minor formatting improvements for better code structure.
3. `basic.py`:
- Introduced a new class `TupleNode` to handle tuple data types.
- Modified `StrBreakout` to include dynamic typing for list conversion.
- Enhanced code readability with better variable naming and added comments for clarity.
4. `pytorch_nodes.py`:
- Added import statements within functions for `torch`.
- New classes `SaveTorchTensor` and `LoadTorchTensor` added for saving and loading torch tensors.
- `GridSearchTraining` class updated with minor formatting changes.
5. `image.py`:
- In `FloodFillNode`, removed a redundant category assignment.
- Added two new classes: `DecorateHeightMap` and `RandomImageFromFolder`. These classes provide functionalities for decorating height maps with images and selecting random images from a folder, respectively.
6. `hf_diffusers_nodes.py`:
- Fixed formatting in tensor format options and added a new line for clarity.
- New class `CreateUnet2dModel` added for creating a UNet2D model with customizable parameters.
- Introduced `TrainUnet2dModel` for training a UNet2D model with a dataset, including detailed setup for the training process.
7. `config.yaml`:
- Added a new base path for `gitio` to centralize the configuration.
These changes aim to improve the maintainability of the codebase, streamline testing processes, and expand the functionalities within the modules, particularly related to image processing and model training in PyTorch and diffusers.
Tasha Upchurch
447e9403ae