put all the functions in model.py

model.py

@@ -10,7 +10,10 @@ from transformers import PretrainedConfig, PreTrainedModel
 import dac
 from audiotools import AudioSignal
 
-from utils import freeze
+
+def freeze(model):
+    for param in model.parameters():
+        param.requires_grad = False
 
 
 class DACConfig(PretrainedConfig):

utils.py

@@ -1,6 +0,0 @@
-
-
-def freeze(model):
-    for param in model.parameters():
-        param.requires_grad = False
-