ceperaltab commited on
Commit
e8876ea
·
verified ·
1 Parent(s): 1ca24b6

Upload train.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +4 -4
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", "your-username/neo4j-dataset")
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", "your-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():