from config import *

def render_condition(
    conf: TrainConfig,
    model,
    sampler, start, motion_direction_start, audio_driven, \
        face_location, face_scale, \
        yaw_pitch_roll, noisyT, control_flag,
):
    if conf.train_mode == TrainMode.diffusion:
        assert conf.model_type.has_autoenc()

        return sampler.sample(model=model,
                              noise=noisyT,
                              model_kwargs={
                                  'motion_direction_start': motion_direction_start, 
                                  'yaw_pitch_roll': yaw_pitch_roll,
                                  'start': start,
                                  'audio_driven': audio_driven,
                                  'face_location': face_location,
                                  'face_scale': face_scale,
                                  'control_flag': control_flag
                              })
    else:
        raise NotImplementedError()