witcherderivia commited on
Commit
705e6ea
·
verified ·
1 Parent(s): 31c3d78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -11
app.py CHANGED
@@ -12,24 +12,22 @@ import os
12
 
13
  from huggingface_hub import hf_hub_download
14
 
15
- from diffsynth import download_customized_models
16
-
17
- download_customized_models(
18
- model_id="Qwen/Qwen-Image-Edit-2509",
19
- local_dir="models/Qwen/Qwen-Image-Edit-2509",
20
- downloading_priority=["HuggingFace"]
21
- )
22
 
23
  pipe = QwenImagePipeline.from_pretrained(
24
  torch_dtype=torch.bfloat16,
25
  device="cuda",
26
  model_configs=[
27
- ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="transformer/diffusion_pytorch_model*.safetensors"),
28
- ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="text_encoder/model*.safetensors"),
29
- ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
 
 
 
 
30
  ],
31
  tokenizer_config=None,
32
- processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="processor/"),
 
33
  )
34
 
35
 
 
12
 
13
  from huggingface_hub import hf_hub_download
14
 
 
 
 
 
 
 
 
15
 
16
  pipe = QwenImagePipeline.from_pretrained(
17
  torch_dtype=torch.bfloat16,
18
  device="cuda",
19
  model_configs=[
20
+ ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
21
+ download_source='huggingface',
22
+ origin_file_pattern="transformer/diffusion_pytorch_model*.safetensors"),
23
+ ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
24
+ download_source='huggingface',origin_file_pattern="text_encoder/model*.safetensors"),
25
+ ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
26
+ download_source='huggingface',origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
27
  ],
28
  tokenizer_config=None,
29
+ processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509",
30
+ download_source='huggingface', origin_file_pattern="processor/"),
31
  )
32
 
33