o
    >	j                  
   @   s   d dl Z d dlmZ d dlmZ dZdZdZdZej	ee j
dd	eZe  eed
ZedZdZeeedddde jedddjd  Zee ede dS )    N)Image)DiffusionPipelinez../input/input1.pngz../output/flux2_output1.pngz&black-forest-labs/FLUX.2-klein-base-9BcudaT)torch_dtypelow_cpu_mem_usageRGB)   r   aK  A highly detailed, realistic fashion catalog product photo of a complete clothing item. Front-view, perfectly symmetrical, fully visible from collar to bottom hem with complete sleeves. Natural fabric folds, preserving the exact original color, texture, and pattern. Plain white studio background, professional lighting, uncropped.r   g      @   )devicei  )imagepromptheightwidthguidance_scalenum_inference_steps	generatorzSaved:)torchPILr   	diffusersr   
INPUT_PATHOUTPUT_PATHMODEL_IDDEVICEfrom_pretrainedfloat16topipeenable_attention_slicingopenconvertr   resizer   	Generatormanual_seedimagesresultsaveprint r'   r'   ;/home/ubuntu/htdocs/OOTD2/inpainting/flux2_klein_base_9b.py<module>   s@   


