Upload model

README.md

@@ -1,6 +1,6 @@
 ---
 library_name: peft
-base_model: facebook/bart-base
+base_model: bigscience/bloomz-560m
 ---
 
 # Model Card for Model ID

adapter_config.json

@@ -1,9 +1,9 @@
 {
   "auto_mapping": null,
-  "base_model_name_or_path": "facebook/bart-base",
+  "base_model_name_or_path": "bigscience/bloomz-560m",
   "inference_mode": true,
-  "num_attention_heads": 12,
-  "num_layers": 6,
+  "num_attention_heads": 16,
+  "num_layers": 24,
   "num_transformer_submodules": 1,
   "num_virtual_tokens": 8,
   "peft_type": "PROMPT_TUNING",
@@ -11,6 +11,6 @@
   "prompt_tuning_init_text": "Create an SQL Query based on the following question and the table structure :",
   "revision": null,
   "task_type": "CAUSAL_LM",
-  "token_dim": 768,
-  "tokenizer_name_or_path": "facebook/bart-base"
+  "token_dim": 1024,
+  "tokenizer_name_or_path": "bigscience/bloomz-560m"
 }

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:036ea92d11f05cac463cc52718f19deb30790fc13147cc0085efba05e579611e
-size 24696
+oid sha256:43508f47f86d2d3763e1ee76d2befa79735bc95caef330255aaad5fa50124d20
+size 32888