alizee-coder-devstral-1-small

prompt = """<s>[INST] Solve this programming problem with detailed reasoning:

Write a function that checks if a number is prime.

[/INST]"""

inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=1024, temperature=0.1)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)

@misc{alizee-coder-devstral,
  author = {stmasson},
  title = {Alizee-Coder-Devstral-1-Small: Code Generation with Reasoning},
  year = {2024},
  publisher = {Hugging Face},
  url = {https://huggingface.co/stmasson/alizee-coder-devstral-1-small}
}