reward-model-score-cuda-device-placement-mixin - Branch - argilla-io/distilabel

Use `CudaDevicePlacementMixin` in `RewardModelScore` step

#845Merged

Comparing

reward-model-score-cuda-device-placement-mixin

(

8cd8002

) with

develop

(

9aa8b35

)

test_cache_time

tests/integration/test_cache.py::test_cache_time

352.5 ms

352.6 ms

Click on a commit to change the comparison range

Base

develop

9aa8b35

Use `CudaDevicePlacementMixin` in `RewardModelScore` step

d356eee

1 year ago

by gabrielmbmb

-7%

Update `_init_cuda_device_placement` to be `LLM` attribute agnostic

3b54165

1 year ago

by gabrielmbmb

+7%

Check if `Step` is instance of `CudaDevicePlacementMixin`

8cd8002

1 year ago

by gabrielmbmb