From 40edb7b5d72a017bfa267668839aceca5123d88f Mon Sep 17 00:00:00 2001
From: binbin Deng <108676127+plusbang@users.noreply.github.com>
Date: Wed, 27 Dec 2023 09:11:37 +0800
Subject: [PATCH] LLM: fix get environment variables setting (#9787)

---
 .../QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning.py    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 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
index 802d8cda..36108cf0 100644
--- 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":