From 78609fb8fd877eb66dc6cedadb6b15793cd0cea6 Mon Sep 17 00:00:00 2001 From: Shubham Ugare Date: Thu, 16 May 2024 09:04:55 -0400 Subject: [PATCH] Update infer.py --- syncode/infer.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/syncode/infer.py b/syncode/infer.py index 7e1eca2..ecd2392 100644 --- a/syncode/infer.py +++ b/syncode/infer.py @@ -37,12 +37,6 @@ class Syncode: log_level (int, optional): Log level. Defaults to 2. 0 for no logs, 1 for minimal logs, 2 for all logs including time. parser (str, optional): Parser to use. Defaults to "lalr". task_id (int, optional): For debugging a specific task. Defaults to None. - - List of currently tested models: - Llama models: "Llama-7b", "CodeLlama-7b", "CodeLlama-7b-Python", "Llama-13b" - CodeGen models: "Salesforce/codegen-350M-multi", "Salesforce/codegen2-1b" - Bigcode models: "bigcode/starcoderbase-1b", "bigcode/santacoder" (1.1b WIP) - WizardLM models: "WizardLM/WizardCoder-1B-V1.0" """ def __init__( self,