o 0yh@snddlZddlZddlZddlZddlmZdejd<ejdej dde dd  d Z e d d d ZdS)N)AutoPipelineForImage2Imagez/data/.cache/huggingfaceHF_HOMEz(stabilityai/stable-diffusion-xl-base-1.0Zfp16THF_TOKEN)Z torch_dtypevariantZuse_safetensorstokencudaztheoracle/sdxl-lora-headshotc Csz1|dur tdtd|j|dd}td|t||p!d|||ddjd }td |WStyN}ztd t t d t |d}~ww) Nz4Uploaded image is None. Please upload a valid image.zNFO] Received image size:RGB)r z%[INFO] Starting pipeline with prompt:2)promptnegative_promptimagestrengthguidance_scalenum_inference_stepsrz[INFO] Generation successful.z)[ERROR] Exception in generate_with_lora: zImage generation failed: ) ValueErrorprintsizeconvertresizepipeimages Exception traceback format_excgrErrorstr)rr r rrresulter!/home/user/app/inference.pygenerate_with_loras.   r#)ostorchrgradior diffusersrenvironfrom_pretrainedfloat16getenvtorZload_lora_weightsr#r!r!r!r"s"