Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
braceal authored Mar 30, 2024
1 parent 80528be commit e4fbf3b
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,6 @@ dataloader = DataLoader(dataset)
embeddings = []
with torch.no_grad():
for batch in dataloader:
for batch in dataloader:
outputs = model(
batch["input_ids"].to(device),
batch["attention_mask"].to(device),
Expand Down

0 comments on commit e4fbf3b

Please sign in to comment.