together default

This commit is contained in:
Xi Yan 2024-11-18 22:39:45 -08:00
parent 7693786322
commit 05e93bd2f7

View file

@ -55,7 +55,7 @@ def get_distribution_template() -> DistributionTemplate:
"inference": [inference_provider], "inference": [inference_provider],
}, },
default_models=default_models, default_models=default_models,
default_shields=[ShieldInput(shield_id="meta-llama/Llama-Guard-3-1B")], default_shields=[ShieldInput(shield_id="meta-llama/Llama-Guard-3-8B")],
), ),
}, },
run_config_env_vars={ run_config_env_vars={