o i, @sddlZddlmZmZmZddlmZddlmZddl m Z gdZ gdZ edd Gd d d Z e Zeed eejd gdeejd gddZe dddedddddgdedddddgddZdZdZe ZdZdZdede fdd Zd!ZdS)"N) dataclassfieldreplace) safe_openMultiModalGuiderParams)SpatioTemporalScaleFactors) ?g?g?gffffff?g333333?皙?333333??)r r r r T)frozenc@seZdZUdZeed<dZeed<dZeed<dZeed<d Z e ed <d Z eed <e d ddZ eed<e dddZeed<edefddZedefddZdS)PipelineParams seedistage_1_heighti stage_1_widthy num_framesg8@ frame_rate(num_inference_stepscCstddddddgdS)N@r ffffff?r cfg_scale stg_scale rescale_scalemodality_scale skip_step stg_blocksrr#r#N/home/ubuntu/LTX-2/packages/ltx-pipelines/src/ltx_pipelines/utils/constants.py#zPipelineParams.)default_factoryvideo_guider_paramscCstddddddgdS)N@r rrrrrrr#r#r#r$r%-r&audio_guider_paramsreturncCt|jdSN)intrselfr#r#r$stage_2_height7zPipelineParams.stage_2_heightcCr,r-)r/rr0r#r#r$ stage_2_width;r3zPipelineParams.stage_2_widthN)__name__ __module__ __qualname__rr/__annotations__rrrrfloatrrr(rr*propertyr2r4r#r#r#r$rs"        r)r")rr(r*i irr g?rr)r )rrrr(r*!z2.3checkpoint_pathr+cCstt}z$t|dd}|pi}Wdn1swY|dd}Wnty:|d|tYSw| t rBt S| d|pHdtS) zDetect pipeline params from checkpoint metadata. Reads the ``model_version`` field from the safetensors config metadata. Returns ``LTX_2_3_PARAMS`` when the version starts with "2.3", otherwise falls back to ``LTX_2_PARAMS``. pt) frameworkN model_versionz@Could not read checkpoint metadata from %s, using LTX-2 defaultsz.Using LTX_2_PARAMS for checkpoint (version=%s)unknown) logging getLoggerr5rmetadataget Exceptionwarning LTX_2_PARAMS startswith_LTX_2_3_MODEL_VERSION_PREFIXLTX_2_3_PARAMSinfo)r@loggerfrHversionr#r#r$ detect_paramshs    rTa6blurry, out of focus, overexposed, underexposed, low contrast, washed out colors, excessive noise, grainy texture, poor lighting, flickering, motion blur, distorted proportions, unnatural skin tones, deformed facial features, asymmetrical face, missing facial features, extra limbs, disfigured hands, wrong hand count, artifacts around text, inconsistent perspective, camera shake, incorrect depth of field, background too sharp, background clutter, distracting reflections, harsh shadows, inconsistent lighting direction, color banding, cartoonish rendering, 3D CGI look, unrealistic materials, uncanny valley effect, incorrect ethnicity, wrong gender, exaggerated expressions, wrong gaze direction, mismatched lip sync, silent or muted audio, distorted voice, robotic voice, echo, background noise, off-sync audio, incorrect dialogue, added dialogue, repetitive speech, jittery movement, awkward pauses, incorrect timing, unnatural transitions, inconsistent framing, tilted camera, flat lighting, inconsistent tone, cinematic oversaturation, stylized filters, or AI artifacts.)rF dataclassesrrr safetensorsrltx_core.components.guidersrltx_core.typesrDISTILLED_SIGMA_VALUESSTAGE_2_DISTILLED_SIGMA_VALUESrrLr(r*rOLTX_2_3_HQ_PARAMSDEFAULT_LORA_STRENGTHDEFAULT_IMAGE_CRFdefaultVIDEO_SCALE_FACTORSVIDEO_LATENT_CHANNELSrNstrrTDEFAULT_NEGATIVE_PROMPTr#r#r#r$sZ   &