Views
No views yet
Salesforce/codegen-350M-multiSalesforce/codegen-350M-multi model. It generates code snippets based on natural language or function signature prompts.Salesforce/codegen-350M-multibitsandbytes (optional)1from datasets import load_dataset
2
3dataset = load_dataset("code_x_glue_cc_code_to_text", "python")1
2from datasets import load_metric
3
4bleu = load_metric("bleu")
5bleu_score = bleu.compute(predictions=["generated_code"], references=["reference_code"])
6print("BLEU Score:", bleu_score)1
2from transformers import pipeline
3
4pipe = pipeline("text-generation", model="./finetuned_codegen_350M", device=0)
5
6prompt = "def is_prime(n):"
7result = pipe(prompt, max_length=100, do_sample=True)
8print(result[0]["generated_text"])
9