Spaces:
Sleeping
Sleeping
Create config.yaml
Browse files- config.yaml +10 -0
config.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
project_name: phi4-training
|
| 2 |
+
model: microsoft/phi-4 # Replace with "microsoft/phi-4" if available
|
| 3 |
+
data_path: ./
|
| 4 |
+
task: text-generation # Or your task (e.g., text-classification)
|
| 5 |
+
learning_rate: 2e-5
|
| 6 |
+
batch_size: 2
|
| 7 |
+
epochs: 3
|
| 8 |
+
fp16: true
|
| 9 |
+
push_to_hub: true
|
| 10 |
+
repo_id: "hackergeek98/phi4-trained"
|