Sijuade commited on
Commit
3cc3906
1 Parent(s): 069dbbe

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +4 -1
utils.py CHANGED
@@ -7,8 +7,11 @@ tokenizer = Config.tokenizer
7
  tokenizer.pad_token = tokenizer.eos_token
8
  tokenizer.add_tokens('<question-answer>')
9
 
 
 
10
 
11
- def prepare_inputs(peft_model, audio_model, clip_model, projection, text_input=None, image_input=None, audio_input=None):
 
12
 
13
  text_audio, text_embed, image_embed = None, None, None
14
 
 
7
  tokenizer.pad_token = tokenizer.eos_token
8
  tokenizer.add_tokens('<question-answer>')
9
 
10
+ peft_model, audio_model = Config.peft_model, Config.audio_model
11
+ clip_model, projection = Config.clip_model, Config.projection
12
 
13
+
14
+ def prepare_inputs(text_input=None, image_input=None, audio_input=None):
15
 
16
  text_audio, text_embed, image_embed = None, None, None
17