fixed?

auto_forge.py

@@ -904,7 +904,7 @@ def start(args) -> float:
         print("Error: No CSV or JSON file given. Please provide one of them.")
         sys.exit(1)
 
-    device = get_device(args)
+    device = torch.device("cpu")
 
     os.makedirs(args.output_folder, exist_ok=True)
 
@@ -1000,7 +1000,7 @@ def start(args) -> float:
     )
 
     # Run optimization loop
-    _run_optimization_loop(optimizer, args, device)
+    _run_optimization_loop(optimizer, args, torch.device("cuda"))
 
     # Post-process, prune, and export outputs
     final_loss = _post_optimize_and_export(
@@ -1014,7 +1014,7 @@ def start(args) -> float:
         material_TDs_np,
         material_names,
         bgr_tuple,
-        device,
+        torch.device("cuda"),
         focus_map_full,
         focus_map_proc,
     )