Model save

README.md

@@ -2,8 +2,6 @@
 license: cc-by-nc-4.0
 library_name: peft
 tags:
-- text-to-audio
-- ylacombe/tiny-punk
 - generated_from_trainer
 base_model: facebook/musicgen-melody
 model-index:
@@ -14,13 +12,9 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/229461195/huggingface/runs/euqah3uh)
 # musicgen-melody-lora-punk
 
-This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on the YLACOMBE/TINY-PUNK - DEFAULT dataset.
-It achieves the following results on the evaluation set:
-- Loss: 4.8576
-- Clap: -0.0187
+This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on an unknown dataset.
 
 ## Model description
 
@@ -40,11 +34,11 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0002
-- train_batch_size: 2
-- eval_batch_size: 1
+- train_batch_size: 16
+- eval_batch_size: 8
 - seed: 456
 - gradient_accumulation_steps: 8
-- total_train_batch_size: 16
+- total_train_batch_size: 128
 - optimizer: Adam with betas=(0.9,0.99) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 4.0
@@ -57,7 +51,7 @@ The following hyperparameters were used during training:
 ### Framework versions
 
 - PEFT 0.11.1
-- Transformers 4.42.0.dev0
-- Pytorch 2.3.0+cu121
-- Datasets 2.19.1
+- Transformers 4.41.2
+- Pytorch 2.3.1+cu121
+- Datasets 2.19.2
 - Tokenizers 0.19.1

adapter_config.json

@@ -23,22 +23,22 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "v_proj",
+    "lm_heads.3",
+    "embed_tokens.3",
     "enc_to_dec_proj",
-    "embed_tokens.0",
-    "embed_tokens.1",
     "out_proj",
+    "embed_tokens.1",
+    "q_proj",
+    "v_proj",
+    "lm_heads.0",
+    "lm_heads.2",
     "k_proj",
-    "embed_tokens.2",
-    "fc2",
     "audio_enc_to_dec_proj",
-    "embed_tokens.3",
-    "q_proj",
-    "lm_heads.3",
     "fc1",
-    "lm_heads.2",
-    "lm_heads.1",
-    "lm_heads.0"
+    "embed_tokens.0",
+    "fc2",
+    "embed_tokens.2",
+    "lm_heads.1"
   ],
   "task_type": null,
   "use_dora": false,

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:26645f3f84341a09d4def575aadedd1c5802e0f9bf10c4c51f5cf7c1c3e62269
+oid sha256:2077f283bdb33ec1aa7357d3956c53a7ed8a5e1a436f59e2039f57b402615798
 size 87103456

config.json

@@ -295,5 +295,5 @@
     "vocab_size": 32128
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.42.0.dev0"
+  "transformers_version": "4.41.2"
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2a79373b974d4c635c296d2fa0ac6290c00cb0e8a357a0bf906d72f554d94851
-size 5304
+oid sha256:9e8a403a0c088d52c06ba1f81d35013c82284c4f340d50697d9dd2041f60d62c
+size 5240