Function to_train_mode

train_eval_vimogen.py:230–232

()

    
    
    def to_train_mode():
        model.train(
        )  # NOTE even when train_target == ['controlnet'] the transformer should be on train mode for training

    def to_eval_mode():
        model.eval()