File size: 761 Bytes
22b462e
 
 
 
 
 
 
 
 
 
 
 
aac688d
22b462e
 
 
 
7546071
22b462e
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import os
import sys

import fire
import gradio as gr

from llama_lora.globals import Global
from llama_lora.ui.main_page import main_page, get_page_title, main_page_custom_css
from llama_lora.utils.data import init_data_dir


def main():
    data_dir = os.path.abspath("./datasets")
    Global.default_base_model_name = "decapoda-research/llama-7b-hf"
    Global.data_dir = data_dir
    Global.load_8bit = False

    Global.ui_dev_mode = False
    Global.ui_dev_mode_title_prefix = ""
    Global.ui_show_sys_info = False

    os.makedirs(data_dir, exist_ok=True)
    init_data_dir()

    with gr.Blocks(title=get_page_title(), css=main_page_custom_css()) as demo:
        main_page()

    demo.queue().launch()


if __name__ == "__main__":
    fire.Fire(main)