Instructions to use PhysicsWallahAI/Aryabhata-1.0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PhysicsWallahAI/Aryabhata-1.0 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="PhysicsWallahAI/Aryabhata-1.0") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("PhysicsWallahAI/Aryabhata-1.0") model = AutoModelForCausalLM.from_pretrained("PhysicsWallahAI/Aryabhata-1.0") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use PhysicsWallahAI/Aryabhata-1.0 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "PhysicsWallahAI/Aryabhata-1.0" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PhysicsWallahAI/Aryabhata-1.0", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/PhysicsWallahAI/Aryabhata-1.0
- SGLang
How to use PhysicsWallahAI/Aryabhata-1.0 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "PhysicsWallahAI/Aryabhata-1.0" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PhysicsWallahAI/Aryabhata-1.0", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "PhysicsWallahAI/Aryabhata-1.0" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PhysicsWallahAI/Aryabhata-1.0", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use PhysicsWallahAI/Aryabhata-1.0 with Docker Model Runner:
docker model run hf.co/PhysicsWallahAI/Aryabhata-1.0
Discussion on Controlling Reasoning Verbosity / Continuation Stability in Aryabhata-1.0
Hi team,
First of all, thank you for open-sourcing Aryabhata-1.0. We have been evaluating the model for educational math-solution generation workflows and found its mathematical reasoning capabilities to be quite strong.
During our experiments, we observed a few architectural behaviors that we wanted to discuss with the community and maintainers to better understand whether these are expected characteristics of the model, or whether there are recommended mitigation strategies.
Observations
Aryabhata appears to be strongly optimized toward detailed reasoning and step-by-step derivations. While this is valuable for deep mathematical explanation, we faced challenges when attempting to generate concise or short-step solutions consistently.
We specifically observed:
- The model tends to generate long Chain-of-Thought (CoT) reasoning by default.
- Prompt tightening alone does not reliably reduce verbosity.
- The reasoning trajectory often expands into exploratory derivations even when concise solutions are requested.
Continuation / Context Window Challenges
In our current inference environment, we are operating under relatively constrained generation windows (~1024 tokens total).
When attempting continuation-based generation for longer solutions, we consistently observed:
- Loss of latent reasoning continuity between generations
- Shift in reasoning trajectory after continuation
- Duplicate or conflicting derivation paths
- Coherence degradation across merged outputs
In many cases:
- the initial generation follows one reasoning path,
- while the continuation begins exploring a different path altogether.
This makes deterministic multi-stage generation difficult for production workflows.
Questions
We wanted to understand from the maintainers/community:
- Is Aryabhata intentionally optimized toward verbose reasoning traces?
- Are there recommended decoding parameters or prompting patterns for obtaining concise solutions?
- Has the team experimented with:
- controllable reasoning depth,
- abbreviated derivation modes,
- or answer-first generation strategies?
- Are continuation-instability behaviors expected for this architecture under constrained context settings?
- Would larger context support significantly improve reasoning continuity?
We would really appreciate any guidance, recommended inference settings, or architectural insights from the team.
Thanks again for releasing the model and making these experiments possible.