Upload train.py with huggingface_hub
Browse files
train.py
CHANGED
|
@@ -31,14 +31,14 @@ load_dotenv()
|
|
| 31 |
# Base model to fine-tune
|
| 32 |
MODEL_NAME = "Qwen/Qwen2.5-Coder-7B-Instruct"
|
| 33 |
|
| 34 |
-
# Dataset - loaded from environment
|
| 35 |
-
DATASET_NAME = os.getenv("HF_DATASET_NAME", "
|
| 36 |
|
| 37 |
# Output directory for the adapter
|
| 38 |
OUTPUT_DIR = "neo4j-cypher-expert"
|
| 39 |
|
| 40 |
-
# Hugging Face Hub settings - loaded from environment
|
| 41 |
-
HF_USERNAME = os.getenv("HF_USERNAME", "
|
| 42 |
|
| 43 |
|
| 44 |
def main():
|
|
|
|
| 31 |
# Base model to fine-tune
|
| 32 |
MODEL_NAME = "Qwen/Qwen2.5-Coder-7B-Instruct"
|
| 33 |
|
| 34 |
+
# Dataset - loaded from environment or use default
|
| 35 |
+
DATASET_NAME = os.getenv("HF_DATASET_NAME", "ceperaltab/neo4j-cypher-dataset")
|
| 36 |
|
| 37 |
# Output directory for the adapter
|
| 38 |
OUTPUT_DIR = "neo4j-cypher-expert"
|
| 39 |
|
| 40 |
+
# Hugging Face Hub settings - loaded from environment or use default
|
| 41 |
+
HF_USERNAME = os.getenv("HF_USERNAME", "ceperaltab")
|
| 42 |
|
| 43 |
|
| 44 |
def main():
|