dangthr commited on
Commit
5dc05fc
·
verified ·
1 Parent(s): 97403e9

Create app3.py

Browse files
Files changed (1) hide show
  1. app3.py +17 -0
app3.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from diffsynth.pipelines.qwen_image import QwenImagePipeline, ModelConfig
2
+ import torch
3
+
4
+
5
+ pipe = QwenImagePipeline.from_pretrained(
6
+ torch_dtype=torch.bfloat16,
7
+ device="cuda",
8
+ model_configs=[
9
+ ModelConfig(model_id="DiffSynth-Studio/Qwen-Image-Distill-Full", origin_file_pattern="diffusion_pytorch_model*.safetensors"),
10
+ ModelConfig(model_id="Qwen/Qwen-Image", origin_file_pattern="text_encoder/model*.safetensors"),
11
+ ModelConfig(model_id="Qwen/Qwen-Image", origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
12
+ ],
13
+ tokenizer_config=ModelConfig(model_id="Qwen/Qwen-Image", origin_file_pattern="tokenizer/"),
14
+ )
15
+ prompt = "精致肖像,水下少女,蓝裙飘逸,发丝轻扬,光影透澈,气泡环绕,面容恬静,细节精致,梦幻唯美。"
16
+ image = pipe(prompt, seed=0, num_inference_steps=15, cfg_scale=1)
17
+ image.save("image.jpg")