Tasha Upchurch
d2d16cadfe
This commit introduces significant enhancements and new features across multiple files in the project. Major changes include: 1. `pytorch_nodes.py`: Expanded PyTorch node functionalities with additional imports, device compatibility checks, optimizer state updates, and training enhancements. Introduced new methods like `optimizer_to` for transferring optimizer state to a specified device, `reset_optimizer_weights` to clear optimizer states, and `make_image_tensor` for converting outputs to image tensors. Also, improved the training process with device compatibility checks, optimizer updates, and the ability to save and load models during training for efficiency. 2. `server_endpoints.py`: Modified server endpoints for improved git integration and user management. Adjustments include dynamic handling of different git repositories, refined branch and user management in GitIO class, and the integration of user-specific git settings in user management functions. 3. `config.yaml`: Added new configurations for temporary directory and git repositories, providing more flexibility and control over file management and version control integration. 4. `torchvision_nodes.py`: Introduced a new file to integrate torchvision functionalities. It includes classes for transform composition, application of torchvision transforms, and handling of torchvision-specific data structures. 5. `hf_diffusers_nodes.py`: Integrated Hugging Face's diffusers library with new nodes for loading and utilizing DDPMPipelines, handling tensor to image conversions, and managing datasets from the Hugging Face ecosystem. 6. `image.py`, `video.py`, `functional.py`: Refactored and enhanced existing functionalities. Included minor tweaks and bug fixes to improve the overall efficiency and robustness of image and video processing capabilities. 7. `requirements.txt`: Updated dependencies, replacing `fastnoisesimd` with `diffusers` to align with the newly integrated Hugging Face diffusers functionalities. 8. `config.py`: Improved the configuration file loading mechanism with dynamic path handling and the ability to replace placeholders with actual paths, enhancing the flexibility of configuration management. This comprehensive update significantly enhances the project's capabilities in machine learning, especially in PyTorch and torchvision, and improves server endpoint functionalities for better user and version control management.
27 lines
767 B
Python
27 lines
767 B
Python
import yaml
|
|
|
|
|
|
def read_yaml(file_path,config_replaclemts):
|
|
with open(file_path, 'r') as file:
|
|
data = yaml.safe_load(file)
|
|
for key, value in data.items():
|
|
if isinstance(value, str):
|
|
for key2, value2 in config_replaclemts.items():
|
|
if key2 in value:
|
|
data[key] = value.replace(key2, value2)
|
|
|
|
return data
|
|
|
|
|
|
import os
|
|
|
|
this_file_path = os.path.dirname(os.path.realpath(__file__))
|
|
# two directories up
|
|
comfy_path = os.path.dirname(os.path.dirname(this_file_path))
|
|
config_replacements = {
|
|
'comfy_path': comfy_path
|
|
}
|
|
config_file_path = os.path.join(this_file_path, "config.yaml")
|
|
# Load the YAML data when the module is imported
|
|
config_settings = read_yaml(config_file_path, config_replacements)
|