Spaces:
Runtime error
Runtime error
zetavg
commited on
fix wandb enable rule
Browse files- 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 = [
|
|
|
48 |
|
49 |
if Config.enable_wandb is None:
|
50 |
-
if
|
51 |
-
Config.
|
52 |
-
|
|
|
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
|