§
    ÇPƒin  ã                   óV   — d dl mZmZ d dlmZmZmZmZmZm	Z	 d dl
mZmZmZmZ g d¢ZdS )é   )ÚPPOStatsÚ
Trajectory)Úestimate_advantagesÚget_reward_penalty_maskÚget_rewards_ppoÚmasked_meanÚ
masked_varÚwhiten)Úget_batch_log_probsÚlogits_to_logprobsÚ%truncate_sequence_at_first_stop_tokenÚtruncate_sequence_for_logprobs)r   r   r   r   r   r   r
   r   r	   r   r   r   N)Ú_typesr   r   Úrewardsr   r   r   r   r	   r
   Úsequence_processingr   r   r   r   Ú__all__© ó    úk/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/rlhf/__init__.pyú<module>r      sÀ   ðð )Ð (Ð (Ð (Ð (Ð (Ð (Ð (ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ðð ð ð ð ð ð ð ð ð ð ð ðð ð €€€r   