mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-20 06:12:07 +00:00
5 lines
109 B
Python
5 lines
109 B
Python
|
import torch
|
||
|
|
|
||
|
|
IS_ROCM_SYSTEM = torch.version.hip is not None
|
||
|
|
IS_CUDA_SYSTEM = torch.version.cuda is not None
|