Hugging Face's logo

Spaces:
Mauricio-100
/
agent-ai
Sleeping

App Files Community
Mauricio-100 commited on
Commit
d808b10
·
verified ·
1 Parent(s): 0db182c

Update scripts/train.py

Browse files
Files changed (1) hide show
  1. scripts/train.py +2 -1
scripts/train.py CHANGED
@@ -1,6 +1,7 @@
1
  from core.model_loader import load_model
2
  from core.tokenizer_loader import load_tokenizer
3
- from core.dataset_loader import load_multiple_datasets
 
4
  from core.trainer import build_trainer
5
 
6
  MODEL_NAME = "tiiuae/falcon-7b-instruct"
 
1
  from core.model_loader import load_model
2
  from core.tokenizer_loader import load_tokenizer
3
+ from core.dataset_loader import load_dataset_single
4
+ dataset = load_dataset_single()
5
  from core.trainer import build_trainer
6
 
7
  MODEL_NAME = "tiiuae/falcon-7b-instruct"