Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Create app3.py
Browse files
    	
        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")
         | 
