Function update_model_config

verl/verl/utils/model.py:61–71

Update the module config with the override_config_kwargs. Args: module_config: The module config from Huggingface Transformers. override_config_kwargs: The kwargs to override the module config.

(module_config, override_config_kwargs)

def update_model_config(module_config, override_config_kwargs):
    """Update the module config with the override_config_kwargs.
    Args:
        module_config: The module config from Huggingface Transformers.
        override_config_kwargs: The kwargs to override the module config.
    """
    for key, val in override_config_kwargs.items():
        if isinstance(val, dict):
            update_model_config(getattr(module_config, key), val)
        else:
            setattr(module_config, key, val)


def get_huggingface_actor_config(model_name: str, override_config_kwargs=None, trust_remote_code=False) -> dict: