├── .editorconfig ├── .github ├── ISSUE_TEMPLATE.md ├── ISSUE_TEMPLATE │ ├── bug-error-report.md │ └── feature_request.md └── workflows │ └── publish_node.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── MANIFEST.in ├── README.md ├── __init__.py ├── example_workflows ├── Chroma_Radiance_Raylight.jpg ├── Chroma_Radiance_Raylight.json ├── Chroma_Raylight.jpg ├── Chroma_Raylight.json ├── FluxDev_Data_Parallel_Raylight.jpg ├── FluxDev_Data_Parallel_Raylight.json ├── FluxDev_Raylight.jpg ├── FluxDev_Raylight.json ├── Flux_2_Edit_Raylight.jpg ├── Flux_2_Edit_Raylight.json ├── Flux_2_GGUF_Raylight.jpg ├── Flux_2_GGUF_Raylight.json ├── Hunyuan_1_5_I2V_Raylight.jpg ├── Hunyuan_1_5_I2V_Raylight.json ├── Hunyuan_Video_Raylight.jpg ├── Kandinsky_I2V_Raylight.jpg ├── Kandinsky_I2V_Raylight.json ├── Qwen_Edit_Raylight.jpg ├── Qwen_Edit_Raylight.json ├── Qwen_Raylight.jpg ├── Qwen_Raylight.json ├── Stable_Diffusion_Raylight.jpg ├── Stable_Diffusion_Raylight.json ├── WanT2V_Raylight.jpg ├── WanT2V_Raylight.json ├── Z_Image_Turbo_Raylight.jpg └── Z_Image_Turbo_Raylight.json ├── icon.png ├── pyproject.toml ├── requirements.txt ├── src └── raylight │ ├── __init__.py │ ├── comfy_dist │ ├── __init__.py │ ├── float.py │ ├── fsdp_registry.py │ ├── lora.py │ ├── model_management.py │ ├── model_patcher.py │ ├── sd.py │ ├── supported_models_base.py │ ├── utils.py │ └── weight_adapter │ │ ├── __init__.py │ │ ├── base.py │ │ ├── loha.py │ │ ├── lokr.py │ │ └── lora.py │ ├── comfy_extra_dist │ ├── __init__.py │ ├── nodes_custom_sampler.py │ ├── nodes_easycache.py │ ├── nodes_model_advanced.py │ ├── nodes_torch_compile.py │ └── ray_patch_decorator.py │ ├── diffusion_models │ ├── __init__.py │ ├── aura │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── chroma │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── chroma_radiance │ │ └── fsdp.py │ ├── cosmos │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── flux │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── hidream │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── hunyuan3d │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── hunyuan3dv2_1 │ │ ├── fsdp.py │ │ └── xdit_context_parallel.py │ ├── hunyuan_video │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── hydit │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── kandinsky5 │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── lightricks │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── lumina │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── modules │ │ └── diffusionmodules │ │ │ └── xdit_cfg_parallel.py │ ├── omnigen │ │ ├── fsdp.py │ │ └── xdit_context_parallel.py │ ├── pixart │ │ ├── fsdp.py │ │ └── xdit_context_parallel.py │ ├── qwen_image │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ └── xdit_context_parallel.py │ ├── utils.py │ └── wan │ │ ├── fsdp.py │ │ ├── xdit_cfg_parallel.py │ │ ├── xdit_context_parallel.py │ │ └── xdit_context_parallel_animate.py │ ├── distributed_modules │ ├── attention.py │ ├── cfg.py │ ├── sageattention_hf_patch.py │ ├── usp.py │ └── utils.py │ ├── distributed_worker │ ├── __init__.py │ ├── parallel_manager.py │ ├── ray_worker.py │ └── utils.py │ ├── expansion │ ├── __init__.py │ ├── comfyui_bnb │ │ └── __init__.py │ └── comfyui_gguf │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ ├── dequant.py │ │ ├── loader.py │ │ ├── nodes.py │ │ └── ops.py │ ├── nodes.py │ └── nodes_debug.py └── tests ├── __init__.py ├── conftest.py ├── pytest.ini └── test_raylight.py /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.editorconfig -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.github/ISSUE_TEMPLATE.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-error-report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.github/ISSUE_TEMPLATE/bug-error-report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/workflows/publish_node.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.github/workflows/publish_node.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/__init__.py -------------------------------------------------------------------------------- /example_workflows/Chroma_Radiance_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Chroma_Radiance_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Chroma_Radiance_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Chroma_Radiance_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Chroma_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Chroma_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Chroma_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Chroma_Raylight.json -------------------------------------------------------------------------------- /example_workflows/FluxDev_Data_Parallel_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/FluxDev_Data_Parallel_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/FluxDev_Data_Parallel_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/FluxDev_Data_Parallel_Raylight.json -------------------------------------------------------------------------------- /example_workflows/FluxDev_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/FluxDev_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/FluxDev_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/FluxDev_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Flux_2_Edit_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Flux_2_Edit_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Flux_2_Edit_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Flux_2_Edit_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Flux_2_GGUF_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Flux_2_GGUF_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Flux_2_GGUF_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Flux_2_GGUF_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Hunyuan_1_5_I2V_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Hunyuan_1_5_I2V_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Hunyuan_1_5_I2V_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Hunyuan_1_5_I2V_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Hunyuan_Video_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Hunyuan_Video_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Kandinsky_I2V_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Kandinsky_I2V_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Kandinsky_I2V_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Kandinsky_I2V_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Qwen_Edit_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Qwen_Edit_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Qwen_Edit_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Qwen_Edit_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Qwen_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Qwen_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Qwen_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Qwen_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Stable_Diffusion_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Stable_Diffusion_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Stable_Diffusion_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Stable_Diffusion_Raylight.json -------------------------------------------------------------------------------- /example_workflows/WanT2V_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/WanT2V_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/WanT2V_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/WanT2V_Raylight.json -------------------------------------------------------------------------------- /example_workflows/Z_Image_Turbo_Raylight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Z_Image_Turbo_Raylight.jpg -------------------------------------------------------------------------------- /example_workflows/Z_Image_Turbo_Raylight.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/example_workflows/Z_Image_Turbo_Raylight.json -------------------------------------------------------------------------------- /icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/icon.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | ray>=2.48.0 2 | xfuser>=0.4.4 3 | kernels 4 | huggingface_hub 5 | hf_transfer -------------------------------------------------------------------------------- /src/raylight/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/comfy_dist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/__init__.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/float.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/float.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/fsdp_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/fsdp_registry.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/lora.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/model_management.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/model_management.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/model_patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/model_patcher.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/sd.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/supported_models_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/supported_models_base.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/utils.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/weight_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/weight_adapter/__init__.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/weight_adapter/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/weight_adapter/base.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/weight_adapter/loha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/weight_adapter/loha.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/weight_adapter/lokr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/weight_adapter/lokr.py -------------------------------------------------------------------------------- /src/raylight/comfy_dist/weight_adapter/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_dist/weight_adapter/lora.py -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/nodes_custom_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_extra_dist/nodes_custom_sampler.py -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/nodes_easycache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_extra_dist/nodes_easycache.py -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/nodes_model_advanced.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_extra_dist/nodes_model_advanced.py -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/nodes_torch_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_extra_dist/nodes_torch_compile.py -------------------------------------------------------------------------------- /src/raylight/comfy_extra_dist/ray_patch_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/comfy_extra_dist/ray_patch_decorator.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/aura/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/aura/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/aura/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/aura/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/chroma/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/chroma/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/chroma/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/chroma/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/chroma/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/chroma/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/chroma_radiance/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/chroma_radiance/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/cosmos/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/cosmos/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/cosmos/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/cosmos/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/cosmos/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/flux/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/flux/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/flux/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/flux/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/flux/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/flux/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hidream/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hidream/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hidream/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hidream/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan3d/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hunyuan3d/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan3d/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hunyuan3d/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan3dv2_1/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan3dv2_1/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan_video/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hunyuan_video/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan_video/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hunyuan_video/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hunyuan_video/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/hunyuan_video/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hydit/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hydit/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/hydit/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/kandinsky5/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/kandinsky5/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/kandinsky5/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/kandinsky5/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/kandinsky5/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lightricks/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lightricks/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/lightricks/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lightricks/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lumina/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/lumina/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lumina/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/lumina/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/lumina/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/lumina/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/modules/diffusionmodules/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/modules/diffusionmodules/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/omnigen/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/omnigen/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/pixart/fsdp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/pixart/xdit_context_parallel.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/diffusion_models/qwen_image/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/qwen_image/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/qwen_image/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/qwen_image/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/qwen_image/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/qwen_image/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/utils.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/wan/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/wan/fsdp.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/wan/xdit_cfg_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/wan/xdit_cfg_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/wan/xdit_context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/wan/xdit_context_parallel.py -------------------------------------------------------------------------------- /src/raylight/diffusion_models/wan/xdit_context_parallel_animate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/diffusion_models/wan/xdit_context_parallel_animate.py -------------------------------------------------------------------------------- /src/raylight/distributed_modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_modules/attention.py -------------------------------------------------------------------------------- /src/raylight/distributed_modules/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_modules/cfg.py -------------------------------------------------------------------------------- /src/raylight/distributed_modules/sageattention_hf_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_modules/sageattention_hf_patch.py -------------------------------------------------------------------------------- /src/raylight/distributed_modules/usp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_modules/usp.py -------------------------------------------------------------------------------- /src/raylight/distributed_modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_modules/utils.py -------------------------------------------------------------------------------- /src/raylight/distributed_worker/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/distributed_worker/parallel_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_worker/parallel_manager.py -------------------------------------------------------------------------------- /src/raylight/distributed_worker/ray_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_worker/ray_worker.py -------------------------------------------------------------------------------- /src/raylight/distributed_worker/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/distributed_worker/utils.py -------------------------------------------------------------------------------- /src/raylight/expansion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/__init__.py -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_bnb/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_bnb/__init__.py -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/.gitignore -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/LICENSE -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/README.md -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/dequant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/dequant.py -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/loader.py -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/nodes.py -------------------------------------------------------------------------------- /src/raylight/expansion/comfyui_gguf/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/expansion/comfyui_gguf/ops.py -------------------------------------------------------------------------------- /src/raylight/nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/nodes.py -------------------------------------------------------------------------------- /src/raylight/nodes_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/src/raylight/nodes_debug.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | """Unit test package for raylight.""" 2 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/tests/pytest.ini -------------------------------------------------------------------------------- /tests/test_raylight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/komikndr/raylight/HEAD/tests/test_raylight.py --------------------------------------------------------------------------------