mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-12 04:44:52 +00:00
Fix nvidia ?
This commit is contained in:
parent
46ccffd246
commit
822131f6dc
@ -506,7 +506,7 @@ def launcher(event_loop):
|
||||
if DOCKER_VOLUME:
|
||||
volumes = [f"{DOCKER_VOLUME}:/data"]
|
||||
|
||||
if DOCKER_DEVICES is not None:
|
||||
if DOCKER_DEVICES:
|
||||
if DOCKER_DEVICES.lower() == "none":
|
||||
devices = []
|
||||
else:
|
||||
|
Loading…
Reference in New Issue
Block a user