networks.py

from config import Config

device = Config.device

audio_model = Config.audio_model
text_model, peft_model = Config.text_model, Config.peft_model
projection, clip_model = Config.projection, Config.clip_model

projection = projection.to(device)
peft_model = peft_model.to(device)
clip_model = clip_model.to(device)