Pi:ddlZddlZdejjddfdZdS)Nmodelreturnc|D]S}t|tjjr2|jdkr't jd|jd|dd|_TdS)z Disables dropout layers in the given model. Args: model (torch.nn.Module): The model in which dropout layers should be disabled. rzFound Dropout with value z in module z. Setting to zero.N)modules isinstancetorchnnDropoutpwarningswarn)rmodules r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/training/_model_util.pydisable_dropoutr s|--// feh. / / FHMM M[FH[[[[[   FH )r rr ModulerrrrsI   58? t      r