From c64e2248efd5a351f1768e2b98bb7dae5808ddd7 Mon Sep 17 00:00:00 2001
From: Heyang Sun <60865256+Uxito-Ada@users.noreply.github.com>
Date: Wed, 13 Dec 2023 11:01:21 +0800
Subject: [PATCH] fix str returned by get_int_from_str rather than expected int
 (#9667)

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

diff --git a/python/llm/example/CPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning_cpu.py b/python/llm/example/CPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning_cpu.py
index 2e26a4e4..93ef8196 100644
--- a/python/llm/example/CPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning_cpu.py
+++ b/python/llm/example/CPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning_cpu.py
@@ -58,7 +58,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 train(
     # model/data params