LLM: fix get environment variables setting (#9787)

2023-12-27 09:11:37 +08:00 · 2023-12-27 09:11:37 +08:00 · 40edb7b5d7
commit 40edb7b5d7
parent 689889482c
1 changed files with 1 additions and 1 deletions
--- a/python/llm/example/GPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning.py
+++ b/python/llm/example/GPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning.py
@ -60,7 +60,7 @@ def get_int_from_env(env_keys, default):
        val = int(os.environ.get(e, -1))
        if val >= 0:
            return val
-    return default
+    return int(default)

 def _get_trainer_cls(training_mode):
    if training_mode == "relora":