state-spaces
/

mamba-370m-hf

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ArthurZ HF staff commited on Mar 6

Commit

14e4f01

•

1 Parent(s): 5891795

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ training_args = TrainingArguments(
 )
 lora_config =  LoraConfig(
         r=8,
-        target_modules="all-linear",
         task_type="CAUSAL_LM",
         bias="none"
 )

 )
 lora_config =  LoraConfig(
         r=8,
+        target_modules=["x_proj", "embeddings", "in_proj", "out_proj"],
         task_type="CAUSAL_LM",
         bias="none"
 )