Skip to content

Commit 57fea7c

Browse files
committed
Remove gpt test
1 parent da63b88 commit 57fea7c

File tree

1 file changed

+0
-22
lines changed

1 file changed

+0
-22
lines changed

tests/test_testable_tasks.py

Lines changed: 0 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -89,25 +89,3 @@ def test_roberta():
8989
)
9090
logits = model(*inputs).logits
9191
logits.sum().backward()
92-
93-
94-
def test_gpt():
95-
model = make_tiny_gpt(seed=0)
96-
dataset = make_gpt_dataset(num_data=8, seed=0)
97-
batch_size = 8
98-
loader = DataLoader(
99-
dataset,
100-
collate_fn=default_data_collator,
101-
batch_size=batch_size,
102-
drop_last=False,
103-
shuffle=False,
104-
)
105-
106-
batch = next(iter(loader))
107-
inputs = (
108-
batch["input_ids"],
109-
None,
110-
batch["attention_mask"],
111-
)
112-
logits = model(*inputs).logits
113-
logits.sum().backward()

0 commit comments

Comments
 (0)