Update app.py

app.py

@@ -1,34 +1,34 @@
-#!/usr/bin/env python3
-"""
-HF Space runner for Wave Tank Surgeon.
-Entry point that executes the wave tank surgeon on DeepSeek-V4-Pro.
-"""
-
-import subprocess
-import sys
-import os
-
-def main():
-    # Ensure HF token is available
-    if not os.getenv('HF_TOKEN'):
-        raise ValueError("HF_TOKEN environment variable not set")
-    
-    # Run the wave tank surgeon
-    cmd = [
-        sys.executable,
-        'wave_tank_surgeon.py',
-        '--model', 'deepseek-ai/DeepSeek-V4-Pro',
-        '--ffn-grid', '16,32,64',
-        '--attn-grid', '8,16,32',
-        '--steps', '100',
-        '--lr', '0.001',
-        '--device', 'cuda:0',
-        '--dtype', 'float16',
-        '--push-to-hub', 'Jibbalit/DeepSeek-V4-Pro-WaveTank',
-    ]
-    
-    result = subprocess.run(cmd, check=True)
-    return result.returncode
-
-if __name__ == '__main__':
-    sys.exit(main())
+#!/usr/bin/env python3
+"""
+HF Space runner for Wave Tank Surgeon.
+Entry point that executes the wave tank surgeon on DeepSeek-V4-Pro.
+"""
+
+import subprocess
+import sys
+import os
+
+def main():
+    # Ensure HF token is available
+    if not os.getenv('HF_TOKEN'):
+        raise ValueError("HF_TOKEN environment variable not set")
+    
+    # Run the wave tank surgeon
+    cmd = [
+        sys.executable,
+        'wave_tank_surgeon.py',
+        '--model', 'deepseek-ai/DeepSeek-V4-Pro',
+        '--ffn-grid', '16,32,64',
+        '--attn-grid', '8,16,32',
+        '--steps', '10000',
+        '--lr', '0.001',
+        '--device', 'cuda:0',
+        '--dtype', 'float16',
+        '--push-to-hub', 'Jibbalit/DeepSeek-V4-Pro-WaveTank',
+    ]
+    
+    result = subprocess.run(cmd, check=True)
+    return result.returncode
+
+if __name__ == '__main__':
+    sys.exit(main())