diff options
Diffstat (limited to 'scripts/test_armo_15666925.out')
| -rw-r--r-- | scripts/test_armo_15666925.out | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/scripts/test_armo_15666925.out b/scripts/test_armo_15666925.out new file mode 100644 index 0000000..7b47c6c --- /dev/null +++ b/scripts/test_armo_15666925.out @@ -0,0 +1,13 @@ +=== Testing ArmoRM Reward Model === +GPU: NVIDIA A100-SXM4-40GB + +====================================================================== +ArmoRM Reward Model Test +====================================================================== +Device: cuda +Model: RLHFlow/ArmoRM-Llama3-8B-v0.1 + +Loading model... +[ArmoRM] Loading model RLHFlow/ArmoRM-Llama3-8B-v0.1 on cuda... + +=== Test Complete === |