From 2ed3860085fdf629e8874444c812e18524b21e38 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Tue, 29 Jul 2025 20:10:08 +0800 Subject: [PATCH] refine code --- examples/flux/model_inference_low_vram/Nexus-Gen-Editing.py | 2 +- examples/flux/model_inference_low_vram/Nexus-Gen-Generation.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/flux/model_inference_low_vram/Nexus-Gen-Editing.py b/examples/flux/model_inference_low_vram/Nexus-Gen-Editing.py index 313ce3c..7dd3193 100644 --- a/examples/flux/model_inference_low_vram/Nexus-Gen-Editing.py +++ b/examples/flux/model_inference_low_vram/Nexus-Gen-Editing.py @@ -22,7 +22,7 @@ pipe = FluxImagePipeline.from_pretrained( ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="text_encoder_2/", offload_device="cpu"), ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="ae.safetensors", offload_device="cpu"), ], - nexus_gen_processor_config=ModelConfig("DiffSynth-Studio/Nexus-GenV2", origin_file_pattern="processor"), + nexus_gen_processor_config=ModelConfig(model_id="DiffSynth-Studio/Nexus-GenV2", origin_file_pattern="processor/"), ) pipe.enable_vram_management() diff --git a/examples/flux/model_inference_low_vram/Nexus-Gen-Generation.py b/examples/flux/model_inference_low_vram/Nexus-Gen-Generation.py index c865271..25feb23 100644 --- a/examples/flux/model_inference_low_vram/Nexus-Gen-Generation.py +++ b/examples/flux/model_inference_low_vram/Nexus-Gen-Generation.py @@ -20,7 +20,7 @@ pipe = FluxImagePipeline.from_pretrained( ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="text_encoder_2/", offload_device="cpu"), ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="ae.safetensors", offload_device="cpu"), ], - nexus_gen_processor_config=ModelConfig("DiffSynth-Studio/Nexus-GenV2", origin_file_pattern="processor"), + nexus_gen_processor_config=ModelConfig(model_id="DiffSynth-Studio/Nexus-GenV2", origin_file_pattern="processor/"), ) pipe.enable_vram_management()