Commit bc97054
[Bugfix][ROCm] Using device_type because on ROCm the API is still torch.cuda (vllm-project#17601)
Signed-off-by: Gregory Shtrasberg <[email protected]>
Signed-off-by: Yuqi Zhang <[email protected]>1 parent bb3a24e commit bc97054
1 file changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
406 | 406 | | |
407 | 407 | | |
408 | 408 | | |
409 | | - | |
| 409 | + | |
410 | 410 | | |
411 | 411 | | |
412 | 412 | | |
413 | 413 | | |
414 | | - | |
| 414 | + | |
415 | 415 | | |
416 | 416 | | |
417 | 417 | | |
| |||
0 commit comments