From 4926a993cf94d03aad4d24be2515bbe70c03c42a Mon Sep 17 00:00:00 2001
From: Carlos Lordelo <carlos.lordelo@epidemicsound.com>
Date: Sun, 6 Oct 2024 19:07:53 +0200
Subject: [PATCH] Add finetune as CLAPAudioConditioner argument

---
 stable_audio_tools/models/conditioners.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/stable_audio_tools/models/conditioners.py b/stable_audio_tools/models/conditioners.py
index e998ab10..a365a3a6 100644
--- a/stable_audio_tools/models/conditioners.py
+++ b/stable_audio_tools/models/conditioners.py
@@ -179,9 +179,11 @@ def __init__(self,
                  clap_ckpt_path,
                  audio_model_type="HTSAT-base", 
                  enable_fusion=True,
-                 project_out: bool = False):
+                 project_out: bool = False,
+                 finetune: bool = False):
         super().__init__(512, output_dim, project_out=project_out)
 
+        self.finetune = finetune
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
         # Suppress logging from transformers