Andrey Vorozhko
commited on
Commit
·
65df198
1
Parent(s):
b650cfe
Import torch in utils functions
Browse files- util_funcs.py +1 -0
util_funcs.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
import random
|
|
|
|
| 2 |
|
| 3 |
def getLengthParam(text: str, tokenizer) -> str:
|
| 4 |
tokens_count = len(tokenizer.encode(text))
|
|
|
|
| 1 |
import random
|
| 2 |
+
import torch
|
| 3 |
|
| 4 |
def getLengthParam(text: str, tokenizer) -> str:
|
| 5 |
tokens_count = len(tokenizer.encode(text))
|