Weak-Driven-Learning

# Clone the repository
git clone https://github.com/chenzehao82/Weak-Driven-Learning.git
cd Weak-Driven-Learning

# Install dependencies
pip install -r requirements.txt

# Install vLLM
pip install 'vllm>=0.11.0'

# Launch OpenAI-compatible API server
vllm serve chhao/Weak-Driven-Learning --port 8000 --tensor-parallel-size 2

from openai import OpenAI

client = OpenAI(
    base_url='http://localhost:8000/v1',
    api_key="EMPTY"
)

messages = [{'role': 'user', 'content': 'Solve: 2x + 3 = 11'}]

completion = client.chat.completions.create(
    messages=messages,
    model="chhao/Weak-Driven-Learning",
    max_tokens=2048,
    temperature=1.0,
    top_p=0.95
)

print(completion.choices[0].message.content)