Skip to content

Commit 5dcb91b

Browse files
committed
do not set HF_HUB_OFFLINE env var for webgui and utility processes
1 parent b98b628 commit 5dcb91b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

self_hosting_machinery/watchdog/docker_watchdog.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,8 @@ def _start(self):
118118
self.set_status("starting")
119119
CUDA_VISIBLE_DEVICES = ",".join(["%d" % x for x in self.cfg["gpus"]])
120120
alt_env["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
121-
if is_hf_hub_offline():
121+
# NOTE: in case CUDA_VISIBLE_DEVICES we assume that process is working with models
122+
if CUDA_VISIBLE_DEVICES and is_hf_hub_offline():
122123
alt_env["HF_HUB_OFFLINE"] = "1"
123124
self.start_ts = time.time()
124125
try:

0 commit comments

Comments
 (0)