"test/git@developer.sourcefind.cn:change/sglang.git" did not exist on "8128e08d3652cef9821aac50a1f2ce99bb282c40"
Unverified Commit b2e1a849 authored by Neo's avatar Neo Committed by GitHub
Browse files

Update dataset_traversal.py

CUDA_VISIBLE_DEVICES 默认值必须为str, 否则split会报错
parent 5ed1e242
......@@ -237,7 +237,7 @@ class SimpleReader(object):
def get_device_num():
if self.use_gpu:
gpus = os.environ.get("CUDA_VISIBLE_DEVICES", 1)
gpus = os.environ.get("CUDA_VISIBLE_DEVICES", '1')
gpu_num = len(gpus.split(','))
return gpu_num
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment