lmzjms commited on
Commit
6316c92
·
1 Parent(s): e154cd6

Update audio_foundation_models.py

Browse files
Files changed (1) hide show
  1. audio_foundation_models.py +1 -1
audio_foundation_models.py CHANGED
@@ -64,7 +64,7 @@ def initialize_model_inpaint(config, ckpt):
64
  sampler = DDIMSampler(model)
65
  return sampler
66
  def select_best_audio(prompt,wav_list):
67
- clap_model = CLAPWrapper('useful_ckpts/CLAP/CLAP_weights_2022.pth','useful_ckpts/CLAP/config.yml',use_cuda=torch.cuda.is_available())
68
  text_embeddings = clap_model.get_text_embeddings([prompt])
69
  score_list = []
70
  for data in wav_list:
 
64
  sampler = DDIMSampler(model)
65
  return sampler
66
  def select_best_audio(prompt,wav_list):
67
+ clap_model = CLAPWrapper('text_to_audio/Make_An_Audio/useful_ckpts/CLAP/CLAP_weights_2022.pth','text_to_audio/Make_An_Audio/useful_ckpts/CLAP/config.yml',use_cuda=torch.cuda.is_available())
68
  text_embeddings = clap_model.get_text_embeddings([prompt])
69
  score_list = []
70
  for data in wav_list: