Skip to content

Commit 2b75949

Browse files
authored
fix: HF env not variables being set when offline mode is False (#43)
1 parent b051f02 commit 2b75949

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

scripts/submit.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,9 +113,10 @@ def main() -> None:
113113
config.slurm.num_nodes = 1
114114
config.slurm.gpus_per_node = 1
115115

116+
if config.container is not None and config.container.env_file:
117+
_apply_hf_env_from_file(config.container.env_file)
118+
116119
if config.offline or config.prefetch_assets:
117-
if config.container is not None and config.container.env_file:
118-
_apply_hf_env_from_file(config.container.env_file)
119120
logger.info(
120121
"offline=True: pre-fetching models and datasets on the login node "
121122
"before submitting the job."

0 commit comments

Comments
 (0)