use_cache=False to avoid warning

midi_model.py

@@ -28,11 +28,13 @@ class MIDIModelConfig:
         net_config = LlamaConfig(vocab_size=tokenizer.vocab_size,
                                  hidden_size=n_embd, num_attention_heads=n_head,
                                  num_hidden_layers=n_layer, intermediate_size=n_inner,
-                                 pad_token_id=tokenizer.pad_id, max_position_embeddings=4096)
+                                 pad_token_id=tokenizer.pad_id, max_position_embeddings=4096,
+                                 use_cache=False)
         net_token_config = LlamaConfig(vocab_size=tokenizer.vocab_size,
                                        hidden_size=n_embd, num_attention_heads=n_head // 4,
                                        num_hidden_layers=n_layer // 4, intermediate_size=n_inner // 4,
-                                       pad_token_id=tokenizer.pad_id, max_position_embeddings=4096)
+                                       pad_token_id=tokenizer.pad_id, max_position_embeddings=4096,
+                                       use_cache=False)
         return MIDIModelConfig(tokenizer, net_config, net_token_config)
 
     @staticmethod