
    hd
                        S SK Jr  SSKJrJrJr  S// SQSS// SQS	.r \" 5       (       d  \" 5       e / S
Q\S'   \(       aL  SSKJr  SSK	J
r
JrJr  SSKJrJr  SSKJrJrJrJrJrJrJr   \" 5       (       d  \" 5       e SS
KJrJrJrJr  gS SKr\" \\" 5       S   \\ S9\RB                  \'   g! \ a     N~f = f! \ a     gf = f)    )TYPE_CHECKING   )OptionalDependencyNotAvailable_LazyModuleis_diffusers_availableget_act_offloading_ctx_manager)GeometricMixtureWrapperPreTrainedModelWrappercreate_reference_model!AutoModelForCausalLMWithValueHead"AutoModelForSeq2SeqLMWithValueHead)SUPPORTED_ARCHITECTURESclone_chat_templateprepare_deepspeedprepare_fsdpprepare_peft_modelsetup_chat_formatunwrap_model_for_generation)activation_offloadingmodeling_basemodeling_value_headutils)DDPOPipelineOutputDDPOSchedulerOutputDDPOStableDiffusionPipeline"DefaultDDPOStableDiffusionPipelinemodeling_sd_base   )r   )r   r   N__file__)module_spec)"typingr   import_utilsr   r   r   _import_structurer   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   sys__name__globals__spec__modules     M/home/james-whalen/.local/lib/python3.13/site-packages/trl/models/__init__.py<module>r,      s    ! ^ ^ ??d?Aef	 !##,.. $
-() Effj  
%''022 (
	
 	
 '')J2GIZhpqCKKO & 		6 * s#   B1 -B= 1B:9B:=CC