AlexWortega commited on
Commit
aea90af
1 Parent(s): e985d60

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -100,6 +100,7 @@ def decode_audio(tokens, quantizer, pad_tokens, n_original_tokens):
100
  # Inference functions
101
  def infer_text_to_audio(text, model, tokenizer, quantizer, max_seq_length=1024, top_k=20):
102
  print(text)
 
103
  text_tokenized = tokenizer(str(text), return_tensors="pt")
104
  text_input_tokens = text_tokenized["input_ids"].to(device)
105
 
 
100
  # Inference functions
101
  def infer_text_to_audio(text, model, tokenizer, quantizer, max_seq_length=1024, top_k=20):
102
  print(text)
103
+ print(type(tokenizer))
104
  text_tokenized = tokenizer(str(text), return_tensors="pt")
105
  text_input_tokens = text_tokenized["input_ids"].to(device)
106