pip install transformers torch

from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("zenlm/zen-coder-flash")
model = AutoModelForCausalLM.from_pretrained("zenlm/zen-coder-flash")

inputs = tokenizer("def fibonacci(n):\n    ", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=128)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

from hanzoai import Hanzo

client = Hanzo(api_key="hk-your-api-key")

response = client.chat.completions.create(
    model="zen-coder-flash",
    messages=[{"role": "user", "content": "Complete this function:\ndef binary_search(arr, target):\n    "}],
)
print(response.choices[0].message.content)