fix str returned by get_int_from_str rather than expected int (#9667)

2023-12-13 11:01:21 +08:00 · 2023-12-13 11:01:21 +08:00 · c64e2248ef
commit c64e2248ef
parent bf1bcf4a14
1 changed files with 1 additions and 1 deletions
--- 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