ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

LLM: fix get environment variables setting (#9787)

Browse source
This commit is contained in:
binbin Deng 2023-12-27 09:11:37 +08:00 committed by GitHub
parent 689889482c
commit 40edb7b5d7
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/example/GPU/QLoRA-FineTuning/alpaca-qlora/alpaca_qlora_finetuning.py
View file

@ -60,7 +60,7 @@ def get_int_from_env(env_keys, default):
val = int(os.environ.get(e, -1)) val = int(os.environ.get(e, -1))
if val >= 0: if val >= 0:
return val return val
return default return int(default)
def _get_trainer_cls(training_mode): def _get_trainer_cls(training_mode):
if training_mode == "relora": if training_mode == "relora":