o jh@s^ddlZddlZddlZddlZdZejddZddZddZe d kr-e edSdS) Nz(stabilityai/stable-diffusion-xl-base-1.0c cs^tjtjttjd}td}td}z_t| dt| dt | t | dVWdn1sFwYWdn1sUwYWtjtjt|dt|dt |t |n!tjtjt|dt|dt |t |wWddS1swYdS)Nw)sysstdoutflushstderropenosdevnulldupdup2fileno contextlibredirect_stdoutredirect_stderrclose)r stdout_fd stderr_fdr6htdocs/OOTD2/inpainting/stable_diffusion_xl_img2img.pysuppress_output s8              "rcCs8d} tj|d|dd}tj|s|S|d7}q)NrToutput_05dz.pngr)r pathjoinexists) output_dirindex output_pathrrrget_next_output_paths r c Cszttjdkr tdtjd}tj|std|tjtj t }tj|}tj t d}tj |d}tj |d}tj |d}tj |d }tj |d }tj |d } ||||||| fD]} tj| d d zt| dWqktyYqkw|tjd<|tjd<|tjd<|tjd<|tjd<|tjd<|tjd<|tjd<|tjd<| tjd<dtjd<dtjd<dtjd<dtjd<tj |d } tj| d d t| } td!dl} d!d"lm}d!d#lm}d!d$lm}d!d$lm}|||t|d%r|| j rd&nd'}|!|"d(}d)}|#d(||fd*}|$d+||j%d}||j&d}|'|||f|j(t)|d&krW| j*n| j+|d&kr`d,ndd |d-,|}|j-d d.|.|d&kr|j/j,| j+d/d0}d1}| j0|d21d3}||||d4d5d6|d7j2d!}|3| Wdn 1swYt4tj5| d d8Wd!St6y}zt4d9|WYd}~dSd}~ww):Nrug입력 파일을 파라메터로 전달하세요. 예: python3 stable_diffusion_xl_img2img.py test1.pngru*입력 파일을 찾을 수 없습니다: z ootd-cache huggingfacehubxetlogstorch matplotlibT)exist_okiHOMEXDG_CACHE_HOMEHF_HOMEHUGGINGFACE_HUB_CACHE HF_HUB_CACHETRANSFORMERS_CACHEDIFFUSERS_CACHE HF_XET_CACHE TORCH_HOME MPLCONFIGDIR1HF_HUB_DISABLE_PROGRESS_BARSerrorTRANSFORMERS_VERBOSITYDIFFUSERS_VERBOSITYfalseTOKENIZERS_PARALLELISMoutputr)Image) StableDiffusionXLImg2ImgPipeline)loggingdisable_progress_barcudacpuRGBi)rArA)rBfp16) torch_dtypevariantuse_safetensors cache_dir)disable)dtypea perfectly symmetrical flat lay clothing, single layer fabric, completely unfolded and spread out, no overlapping parts, straight-on orthographic view, dead center, strictly parallel vertical lines, perfectly horizontal hem, smooth ironed fabric, no wrinkles, professional studio photography, white background, high quality catalog photo z overlapping fabric, folded layers, crumpled, double layers, tilted, asymmetrical, distorted, curved lines, warped, mannequin, body, person, blurry, low quality )device*g333333?g&@2)promptnegative_promptimagestrengthguidance_scalenum_inference_steps generator)rzERROR: )7lenrargv ValueErrorr risfileFileNotFoundErrordirnameabspath__file__rtempfile gettempdirmakedirschmodPermissionErrorenvironr rr%PILr: diffusersr;diffusers.utilsr<transformers.utilsset_verbosity_errorr=hasattrr> is_availabler convertnew thumbnailwidthheightpastefrom_pretrainedMODEL_IDfloat16float32toset_progress_bar_configenable_attention_slicingvae Generator manual_seedimagessaveprintbasename Exception) input_path script_dir parent_dirrGhuggingface_dirhuggingface_hub_dirhuggingface_xet_dirhuggingface_xet_logs_dir torch_dirmatplotlib_dir directoryrrr%r:r;diffusers_loggingtransformers_loggingrJrO canvas_sizecanvasxypiperMrNrSresultexcrrrmain's                             Ur__main__) r rr\rrpcontextmanagerrr r__name__exitrrrrs