Views
No views yet
1! pip install transformers accelerate --quiet
2from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer, AutoConfig
3
4tokenizer = AutoTokenizer.from_pretrained("codellama/CodeLlama-7b-Instruct-hf")
5model = AutoModelForCausalLM.from_pretrained("monsoon-nlp/codellama-abliterated-2xd", device_map="auto")
6
7code_generator = pipeline('text-generation', model=model, tokenizer=tokenizer, do_sample=False)
8
9input_string = "[INST] Write a python function to calculate the factorial of a number [/INST]"
10generated_code = code_generator(input_string, max_length=100)[0]['generated_text']
11print(generated_code)