Commit f240c79
[Bugfix][ROCm] Using device_type because on ROCm the API is still torch.cuda (vllm-project#17601)
Signed-off-by: Gregory Shtrasberg <[email protected]>1 parent c0a2073 commit f240c79
1 file changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
413 | 413 | | |
414 | 414 | | |
415 | 415 | | |
416 | | - | |
| 416 | + | |
417 | 417 | | |
418 | 418 | | |
419 | 419 | | |
420 | 420 | | |
421 | | - | |
| 421 | + | |
422 | 422 | | |
423 | 423 | | |
424 | 424 | | |
| |||
0 commit comments