HelpingAI-Lite is a lite version of the HelpingAI model that can assist with coding tasks. It's trained on a diverse range of datasets and fine-tuned to provide accurate and helpful responses.
License
This model is licensed under MIT.
Datasets
The model was trained on the following datasets:
cerebras/SlimPajama-627B
bigcode/starcoderdata
HuggingFaceH4/ultrachat_200k
HuggingFaceH4/ultrafeedback_binarized
Language
The model supports English language.
Usage
CPU and GPU code
fromtransformersimportpipelinefromaccelerateimportAccelerator# Initialize the acceleratoraccelerator=Accelerator()# Initialize the pipelinepipe=pipeline("text-generation",model="OEvortex/HelpingAI-Lite",device=accelerator.device)# Define the messagesmessages=[{"role":"system","content":"You are a chatbot who can help code!",},{"role":"user","content":"Write me a function to calculate the first 10 digits of the fibonacci sequence in Python and print it out to the CLI.",},]# Prepare the promptprompt=pipe.tokenizer.apply_chat_template(messages,tokenize=False,add_generation_prompt=True)# Generate predictionsoutputs=pipe(prompt,max_new_tokens=256,do_sample=True,temperature=0.7,top_k=50,top_p=0.95)# Print the generated textprint(outputs[0]["generated_text"])