o Ti* @s*ddlmZddlmZmZmZdggdddggdd Zzes%eWn ey/Ynwgd ed <erd d lmZd dl m Z m Z m Z d dl mZmZd dlmZmZmZmZmZmZzesgeWn eyrYdSwd d lmZmZmZmZdSddlZeeedeedej e<dS)) TYPE_CHECKING)OptionalDependencyNotAvailable _LazyModuleis_diffusers_availableget_act_offloading_ctx_manager)GeometricMixtureWrapperPreTrainedModelWrappercreate_reference_model!AutoModelForCausalLMWithValueHead"AutoModelForSeq2SeqLMWithValueHead)SUPPORTED_ARCHITECTURESclone_chat_templateprepare_deepspeed prepare_fsdpsetup_chat_formatunwrap_model_for_generation)activation_offloading modeling_basemodeling_value_headutils)DDPOPipelineOutputDDPOSchedulerOutputDDPOStableDiffusionPipeline"DefaultDDPOStableDiffusionPipelinemodeling_sd_base)r)r r N__file__) module_spec)!typingr import_utilsrrr_import_structurerrrrr r rr r rr rrrrrrrrrrsys__name__globals__spec__modulesr'r'G/home/ubuntu/.local/lib/python3.10/site-packages/trl/models/__init__.pys<