tests/samplers/test_rejection_sampler.py · e67c295b0c812a4969d5bbc70818f6d87c87f9e4 · OpenDAS / vllm_cscc · GitLab

"examples/llm/components/dyn:/nemotron_ultra.backend.generate" did not exist on "aeb79e62775d7436fb14fe45acc75c64c9e2526e"

Find file Blame History Permalink

Remove hardcoded `device="cuda" ` to support more devices (#2503) · 96b6f475
Kunshang Ji authored Feb 02, 2024
```
Co-authored-by: Jiang Li <jiang1.li@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
```
96b6f475

test_rejection_sampler.py 15.1 KB