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:
|
if DOCKER_VOLUME:
|
||||||
volumes = [f"{DOCKER_VOLUME}:/data"]
|
volumes = [f"{DOCKER_VOLUME}:/data"]
|
||||||
|
|
||||||
if DOCKER_DEVICES is not None:
|
if DOCKER_DEVICES:
|
||||||
if DOCKER_DEVICES.lower() == "none":
|
if DOCKER_DEVICES.lower() == "none":
|
||||||
devices = []
|
devices = []
|
||||||
else:
|
else:
|
||||||
|
Loading…
Reference in New Issue
Block a user