We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b98b628 commit 5dcb91bCopy full SHA for 5dcb91b
self_hosting_machinery/watchdog/docker_watchdog.py
@@ -118,7 +118,8 @@ def _start(self):
118
self.set_status("starting")
119
CUDA_VISIBLE_DEVICES = ",".join(["%d" % x for x in self.cfg["gpus"]])
120
alt_env["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
121
- if is_hf_hub_offline():
+ # NOTE: in case CUDA_VISIBLE_DEVICES we assume that process is working with models
122
+ if CUDA_VISIBLE_DEVICES and is_hf_hub_offline():
123
alt_env["HF_HUB_OFFLINE"] = "1"
124
self.start_ts = time.time()
125
try:
0 commit comments