Spaces:
Build error
Build error
| import os | |
| import requests | |
| import json | |
| from pathlib import Path | |
| # Use the environment variables from the task | |
| token = os.environ.get("HF_TOKEN") | |
| namespace = "GAInTech" | |
| space_id = "GAInTech/feather-a10g-pretrain-v2-1778241600" | |
| flavor = "a10g-large" | |
| # Constructed environment - minimal for testing first | |
| env = { | |
| "FEATHER_RUNTIME_MODE": "job", | |
| "HYDRA_BATCH_SIZE": "96", | |
| "HYDRA_TOTAL_BATCH": "196608", | |
| "HYDRA_USE_NEMOTRON": "1", | |
| "HYDRA_TARGET_SHARDS": "0", | |
| "HYDRA_FORCE_HTM_CPU": "1", | |
| "HYDRA_INERT_MAMBA": "1" | |
| } | |
| # The payload that actually works for space-backed jobs | |
| payload = { | |
| "spaceId": space_id, | |
| "command": ["/bin/bash", "-c", "python /app/entrypoint.py"], | |
| "env": env, | |
| "secrets": {"HF_TOKEN": token}, | |
| "flavor": flavor, | |
| "timeout": "12h" | |
| } | |
| url = f"https://huggingface.co/api/jobs/{namespace}" | |
| headers = { | |
| "Authorization": f"Bearer {token}", | |
| "Content-Type": "application/json" | |
| } | |
| print(f"POST {url}") | |
| # Sending with spaceId (CamelCase) | |
| resp = requests.post(url, json=payload, headers=headers) | |
| print(f"Status: {resp.status_code}") | |
| print(f"Response: {resp.text}") | |
| if resp.status_code != 200: | |
| # Fallback to space_id (snake_case) | |
| payload.pop("spaceId") | |
| payload["space_id"] = space_id | |
| print("Retrying with space_id...") | |
| resp = requests.post(url, json=payload, headers=headers) | |
| print(f"Status: {resp.status_code}") | |
| print(f"Response: {resp.text}") | |