fix axolotl example

Author: nielsrolf

example/axolotl_example/start_job.py

@@ -7,6 +7,6 @@
 job = ow.axolotl.create(
     local_config_dir=os.path.dirname(__file__),
     command='axolotl train llama_3_70b_fft.yaml',
-    allowed_hardware=['8x H100']
+    allowed_hardware=['8x H100', '8x A100', '8x H100S', '8x H100N', '8x A100S']
 )
 print(job)

example/axolotl_example/zero1.json

@@ -0,0 +1,23 @@
+{
+    "zero_optimization": {
+      "stage": 1,
+      "overlap_comm": true
+    },
+    "bf16": {
+      "enabled": "auto"
+    },
+    "fp16": {
+      "enabled": "auto",
+      "auto_cast": false,
+      "loss_scale": 0,
+      "initial_scale_power": 32,
+      "loss_scale_window": 1000,
+      "hysteresis": 2,
+      "min_loss_scale": 1
+    },
+    "gradient_accumulation_steps": "auto",
+    "gradient_clipping": "auto",
+    "train_batch_size": "auto",
+    "train_micro_batch_size_per_gpu": "auto",
+    "wall_clock_breakdown": false
+  }