mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-25 21:57:45 +00:00
added a comment in the tests & re run tests upstream, works locally
This commit is contained in:
parent
736404c1bd
commit
41a45580e0
1 changed files with 2 additions and 0 deletions
|
@ -46,6 +46,7 @@ sys.stdout.reconfigure(line_buffering=True)
|
|||
# -v -s --tb=short --disable-warnings
|
||||
|
||||
|
||||
# SFT test
|
||||
class TestPostTraining:
|
||||
@pytest.mark.integration
|
||||
@pytest.mark.parametrize(
|
||||
|
@ -155,6 +156,7 @@ class TestPostTraining:
|
|||
# assert job_artifacts.checkpoints[0].epoch == 0
|
||||
# assert "/.llama/checkpoints/Llama3.2-3B-Instruct-sft-0" in job_artifacts.checkpoints[0].path
|
||||
|
||||
# DPO test
|
||||
@pytest.mark.integration
|
||||
@pytest.mark.parametrize(
|
||||
"purpose, source",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue