mhhmm
/

codegen-6B-lora

Text Generation

Inference Endpoints

Model card Files Files and versions Community

mhhmm commited on Apr 26, 2023

Commit

ac49971

•

1 Parent(s): afd82ee

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -3,7 +3,7 @@
   "peft_model_id": "mhhmm/codegen-6B-LoRA",
   "_name_or_path": "codegen-6B-mono",
   "architectures": [
-    "AutoModelForCausalLM"
   ],
   "tokenizer_class": "GPT2Tokenizer",
   "attention_probs_dropout_prob": 0.0,

   "peft_model_id": "mhhmm/codegen-6B-LoRA",
   "_name_or_path": "codegen-6B-mono",
   "architectures": [
+    "CodeGenForCausalLM"
   ],
   "tokenizer_class": "GPT2Tokenizer",
   "attention_probs_dropout_prob": 0.0,