| import os |
|
|
| BASE_DIR = "/home/adminuser/aiops_pocs/MAMBA_FINETUNING" |
| DATA_DIR = os.path.join(BASE_DIR, "data") |
| RESULTS_DIR = os.path.join(BASE_DIR, "results") |
| DOCS_DIR = os.path.join(BASE_DIR, "docs") |
|
|
| MODEL_ID = "tiiuae/falcon-mamba-7b-instruct" |
| DATASET_ID = "BespokeLabs/Bespoke-Stratos-17k" |
|
|
| HF_HUB_REPO = "namanadep/Mamba-7B-Reasoning" |
| OLLAMA_MODEL_NAME = "mamba-7b-reasoning:latest" |
|
|
| TRAIN_FILE = os.path.join(DATA_DIR, "train.jsonl") |
| VAL_FILE = os.path.join(DATA_DIR, "val.jsonl") |
| EVAL_FILE = os.path.join(DATA_DIR, "eval_100_prompts.jsonl") |
|
|
| OUTPUT_ADAPTER_DIR = os.path.join(RESULTS_DIR, "adapter") |
| OUTPUT_MERGED_DIR = os.path.join(RESULTS_DIR, "merged") |
| GGUF_OUTPUT_FILE = os.path.join(RESULTS_DIR, "mamba-7b-reasoning-f16.gguf") |
| EVAL_RESULTS_JSON = os.path.join(RESULTS_DIR, "eval_results.json") |
| EVAL_COMPARISON_MD = os.path.join(DOCS_DIR, "MAMBA_7B_REASONING_BENCHMARK_REPORT.md") |
|
|
