Spaces:
Paused
Paused
File size: 911 Bytes
671ce97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import os
import sys
from antigravity_sdk.client import RemoteGPU
# Config
SCRIPT_PATH = "examples/deep_nanogpt.py"
def main():
if not os.path.exists(SCRIPT_PATH):
print(f"❌ Script not found: {SCRIPT_PATH}")
sys.exit(1)
with open(SCRIPT_PATH, 'r') as f:
code = f.read()
print(f"🚀 Launching Deep-NanoGPT Experiment ({len(code)} bytes)...")
gpu = RemoteGPU()
# Run synchronously (blocking) as streaming is disabled
result = gpu.run(code, download_files=False)
print("\n" + "="*50)
print("OUTPUT REMOTO:")
print("="*50)
print(result.output)
# Check for downloaded files
if os.path.exists("comparison_loss.png"):
print("\n✅ Grafico salvato: comparison_loss.png")
if os.path.exists("generation_sample.txt"):
print("✅ Sample salvato: generation_sample.txt")
if __name__ == "__main__":
main()
|