Spaces:
Running
Running
AlexWortega
commited on
Commit
•
aea90af
1
Parent(s):
e985d60
Update main.py
Browse files
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 |
|