Function to_eval_mode

train_eval_vimogen.py:234–235

()

        )  # NOTE even when train_target == ['controlnet'] the transformer should be on train mode for training

    def to_eval_mode():
        model.eval()
    
    wan_scheduler = FlowMatchScheduler()