Upload modeling_tara.py with huggingface_hub
Browse files- modeling_tara.py +1 -1
modeling_tara.py
CHANGED
|
@@ -374,7 +374,7 @@ if __name__ == "__main__":
|
|
| 374 |
|
| 375 |
# Let's encode a sample text
|
| 376 |
print(colored("Testing text encoding...", 'cyan'))
|
| 377 |
-
text = ['someone is folding a paper', 'cutting a paper', 'someone is
|
| 378 |
# NOTE: It can also take a single string
|
| 379 |
with torch.no_grad():
|
| 380 |
text_emb = model.encode_text(text).cpu().float()
|
|
|
|
| 374 |
|
| 375 |
# Let's encode a sample text
|
| 376 |
print(colored("Testing text encoding...", 'cyan'))
|
| 377 |
+
text = ['someone is folding a paper', 'cutting a paper', 'someone is unfolding a paper']
|
| 378 |
# NOTE: It can also take a single string
|
| 379 |
with torch.no_grad():
|
| 380 |
text_emb = model.encode_text(text).cpu().float()
|