zetavg commited on
Commit
438ee0e
·
unverified ·
1 Parent(s): 28664d2

fix wandb enable rule

Browse files
Files changed (1) hide show
  1. llama_lora/config.py +6 -4
llama_lora/config.py CHANGED
@@ -44,10 +44,12 @@ def process_config():
44
  Config.timezone = pytz.timezone(Config.timezone)
45
 
46
  if Config.default_base_model_name not in Config.base_model_choices:
47
- Config.base_model_choices = [Config.default_base_model_name] + Config.base_model_choices
 
48
 
49
  if Config.enable_wandb is None:
50
- if Config.wandb_api_key and len(Config.wandb_api_key) > 0:
51
- Config.enable_wandb = True
52
- if Config.default_wandb_project and len(Config.default_wandb_project) > 0:
 
53
  Config.enable_wandb = True
 
44
  Config.timezone = pytz.timezone(Config.timezone)
45
 
46
  if Config.default_base_model_name not in Config.base_model_choices:
47
+ Config.base_model_choices = [
48
+ Config.default_base_model_name] + Config.base_model_choices
49
 
50
  if Config.enable_wandb is None:
51
+ if (
52
+ Config.wandb_api_key and len(Config.wandb_api_key) > 0
53
+ and Config.default_wandb_project and len(Config.default_wandb_project) > 0
54
+ ):
55
  Config.enable_wandb = True