blob: 7b47c6c8138e05cee388623eb3431dc0b2e0339f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
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 ===
|