Skip to content

Commit ba34e71

Browse files
committed
remove diffs
1 parent 5baa2c0 commit ba34e71

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

bigcode_eval/generation.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ def __call__(self, input_ids, scores, **kwargs):
3838
"""Returns true if generated sequence is too long."""
3939
return input_ids.shape[1] > int(self.input_length * self.multiplier)
4040

41-
4241
def parallel_generations(
4342
task,
4443
dataset,
@@ -108,7 +107,6 @@ def parallel_generations(
108107
print(f"number of problems for this task is {n_tasks}")
109108
n_copies = ceil(args.n_samples / args.batch_size)
110109

111-
print("TokenizedDataset...")
112110
ds_tokenized = TokenizedDataset(
113111
task,
114112
dataset,
@@ -138,7 +136,7 @@ def parallel_generations(
138136
else:
139137
# model.to() is not supported for 8bit and 4bit models
140138
model, ds_loader = accelerator.prepare(model, ds_loader)
141-
139+
142140
generations = complete_code(
143141
task,
144142
accelerator,

0 commit comments

Comments
 (0)