Commit graph

39 commits

Author SHA1 Message Date
Connor Hack
1912ff2341 Temporarily make repo point to fork for PR testing 2024-11-25 07:30:51 -08:00
Connor Hack
cd0c80d61f Add env vars debug printout 2024-11-25 07:13:54 -08:00
Connor Hack
e428b82398 Revert test formula 2024-11-25 06:53:37 -08:00
Chacksu
217f81bdb1
Merge branch 'meta-llama:main' into main 2024-11-25 09:38:35 -05:00
Ashwin Bharambe
31e983ab68 Simplify feature request ISSUE template 2024-11-22 18:02:39 -08:00
Connor Hack
071710426d Try shortening test formula 2024-11-22 15:42:55 -08:00
Connor Hack
8f60a3a55d Clean up job names 2024-11-22 15:07:08 -08:00
Connor Hack
cbd69d06c3 Clean up checkpoint directory setting 2024-11-22 14:22:31 -08:00
Connor Hack
d1d8f859e6 Update checkpointd directory setting 2024-11-22 12:51:34 -08:00
Connor Hack
7f5e0dd3db Refactor test run to support shorthand model names 2024-11-22 12:30:13 -08:00
Connor Hack
9c07e0189a Fix syntax error 2024-11-22 11:16:17 -08:00
Connor Hack
0e9ed3688d Remove unnecessary env vars 2024-11-22 10:58:17 -08:00
Connor Hack
1481a67365 Test new provider name 2024-11-22 10:22:12 -08:00
Connor Hack
143e91f23d Add manual provider back for testing 2024-11-22 09:18:29 -08:00
Connor Hack
25e23a1dfe Add debug statement for PROVIDER_ID 2024-11-22 08:56:53 -08:00
Connor Hack
496879795e Dynamically change provider in tests 2024-11-22 07:22:04 -08:00
Connor Hack
046eec9793 Remove testing llama-stack RC 2024-11-21 16:35:00 -08:00
Connor Hack
318c98807c Pre-emptively test llama stack RC 2024-11-21 15:15:43 -08:00
Connor Hack
16ffe19a20 Account for if a permitted model is None 2024-11-20 18:48:59 -08:00
Connor Hack
2795731434 Update model name for mete-reference template 2024-11-20 14:40:37 -08:00
Chacksu
edfd92d81f
Merge branch 'meta-llama:main' into main 2024-11-20 16:12:38 -05:00
Ashwin Bharambe
983d6ce2df
Remove the "ShieldType" concept (#430)
# What does this PR do?

This PR kills the notion of "ShieldType". The impetus for this is the
realization:

> Why is keyword llama-guard appearing so many times everywhere,
sometimes with hyphens, sometimes with underscores?

Now that we have a notion of "provider specific resource identifiers"
and "user specific aliases" for those and the fact that this works with
models ("Llama3.1-8B-Instruct" <> "fireworks/llama-3pv1-..."), we can
follow the same rules for Shields.

So each Safety provider can make up a notion of identifiers it has
registered. This already happens with Bedrock correctly. We just
generalize it for Llama Guard, Prompt Guard, etc.

For Llama Guard, we further simplify by just adopting the underlying
model name itself as the identifier! No confusion necessary.

While doing this, I noticed a bug in our DistributionRegistry where we
weren't scoping identifiers by type. Fixed.

## Feature/Issue validation/testing/test plan

Ran (inference, safety, memory, agents) tests with ollama and fireworks
providers.
2024-11-12 12:37:24 -08:00
Connor Hack
69215b6f6c Update checkpoint directory to include model 2024-11-07 06:36:27 -08:00
Chacksu
b7f18eb012
Merge branch 'meta-llama:main' into main 2024-11-07 09:33:47 -05:00
Ashwin Bharambe
cfcc0a871c Slightly update PR template 2024-11-06 22:49:01 -08:00
Connor Hack
91b37c8972 Remove old environment variables 2024-11-06 13:24:15 -08:00
Connor Hack
00829b8537 Rename checkpoint directory env var 2024-11-06 12:52:07 -08:00
Connor Hack
f6b658e8b9 Add torchrun specially for meta_reference 2024-11-06 11:12:55 -08:00
Connor Hack
274b5a0d0c Add WORLD_SIZE env var for torch.distributed 2024-11-06 10:16:19 -08:00
Connor Hack
307b2ca33e Add RANK environment variable for torch.distributed 2024-11-06 09:58:35 -08:00
Connor Hack
e200a4232c Add aiosqlite as library dependency 2024-11-06 09:38:42 -08:00
Connor Hack
02663d86b5 Remove Workplace Message notification Action 2024-11-06 09:25:10 -08:00
Connor Hack
9d5636180c Update torchrun to pytest and checkpoint directory env var 2024-11-06 09:23:57 -08:00
Connor Hack
e972a04b7d Add workflow for GitHub Actions 2024-11-06 09:17:26 -08:00
Justin Lee
b6d8246b82
added templates and enhanced readme (#307)
Co-authored-by: Justin Lee <justinai@fb.com>
2024-10-24 17:07:06 -07:00
Yuan Tang
a2b87ed0cb
Switch to pre-commit/action (#239) 2024-10-11 11:09:11 -07:00
Yuan Tang
05282d1234
Enable pre-commit on main branch (#237) 2024-10-11 10:03:59 -07:00
Russell Bryant
eba9d1ea14
ci: Run pre-commit checks in CI (#176)
Run the pre-commit checks in a github workflow to validate that a PR
or a direct push to the repo does not introduce new errors.
2024-10-10 11:21:59 -07:00
Xi Yan
1433aaf9f7 add CODEOWNERS file 2024-09-11 11:40:37 -07:00