all the custom functions must be in the same file

model.py

@@ -10,14 +10,17 @@ 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):
     model_type = 'dac'
 
     def __init__(self, 
-                 model_type_by_sampling_freq:str='44khz',
+                 model_type_by_sampling_freq:str='16khz',
                  encoding_chunk_size_in_sec:int=1,
                  decoding_chunk_rate:float=0.1,
                  decoding_overlap_rate:float=0.1,

utils.py

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