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":