bpiyush commited on
Commit
daa775c
·
verified ·
1 Parent(s): 8e13a12

Upload modeling_tara.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 folding 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()
 
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()