Spaces:
Running
Running
remove unneeded comment
Browse files
app.py
CHANGED
@@ -52,7 +52,6 @@ def get_spans_local(prompt, doc, updated_doc):
|
|
52 |
updated_doc_ids = tokenizer(updated_doc, return_tensors='pt')['input_ids'][0]
|
53 |
joined_ids = torch.cat([tokenized_chat, updated_doc_ids[1:]])
|
54 |
|
55 |
-
# Call the model
|
56 |
with torch.no_grad():
|
57 |
logits = model(joined_ids[None].to(model.device)).logits[0].cpu()
|
58 |
|
|
|
52 |
updated_doc_ids = tokenizer(updated_doc, return_tensors='pt')['input_ids'][0]
|
53 |
joined_ids = torch.cat([tokenized_chat, updated_doc_ids[1:]])
|
54 |
|
|
|
55 |
with torch.no_grad():
|
56 |
logits = model(joined_ids[None].to(model.device)).logits[0].cpu()
|
57 |
|