some fixes

llama_lora/globals.py

@@ -40,6 +40,7 @@ class Global:
     ui_subtitle: str = "Toolkit for evaluating and fine-tuning LLaMA models with low-rank adaptation (LoRA)."
     ui_show_sys_info: bool = True
     ui_dev_mode: bool = False
+    ui_dev_mode_title_prefix: str = "[UI DEV MODE] "
 
 
 def get_package_dir():

llama_lora/models.py

@@ -98,7 +98,7 @@ def load_base_model():
             )
         else:
             Global.loaded_base_model = LlamaForCausalLM.from_pretrained(
-                base_model, device_map={"": device}, low_cpu_mem_usage=True
+                Global.base_model, device_map={"": device}, low_cpu_mem_usage=True
             )
 
 

llama_lora/ui/main_page.py

@@ -69,7 +69,7 @@ def main_page():
 def get_page_title():
     title = Global.ui_title
     if (Global.ui_dev_mode):
-        title = f"[UI DEV MODE] {title}"
+        title = Global.ui_dev_mode_title_prefix + title
     if (Global.ui_emoji):
         title = f"{Global.ui_emoji} {title}"
     return title