theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #884 from modelscope/dev2-dzj

Browse Source 
Unified Dataset & Splited Training
This commit is contained in:
Zhongjie Duan
2025-09-03 09:50:23 +08:00
committed by GitHub
parent c13fd7e0ee 958ebf1352
commit 8c13362dcf
8 changed files with 598 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
diffsynth/pipelines/qwen_image.py
View File

@@ -174,9 +174,12 @@ class QwenImagePipeline(BasePipeline):
computation_dtype=self.torch_dtype,
computation_device="cuda",
)
enable_vram_management(self.text_encoder, module_map=module_map, module_config=model_config)
enable_vram_management(self.dit, module_map=module_map, module_config=model_config)
enable_vram_management(self.vae, module_map=module_map, module_config=model_config)
if self.text_encoder is not None:
enable_vram_management(self.text_encoder, module_map=module_map, module_config=model_config)
if self.dit is not None:
enable_vram_management(self.dit, module_map=module_map, module_config=model_config)
if self.vae is not None:
enable_vram_management(self.vae, module_map=module_map, module_config=model_config)
def enable_vram_management(self, num_persistent_param_in_dit=None, vram_limit=None, vram_buffer=0.5, enable_dit_fp8_computation=False):