Skip to content

Commit d29f7eb

Browse files
Indraneil PaulIndraneil Paul
Indraneil Paul
authored and
Indraneil Paul
committed
Minor fixes
1 parent 627aa5d commit d29f7eb

3 files changed

Lines changed: 0 additions & 19 deletions

File tree

‎Docker/Generate_Cuda11.Dockerfile‎

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -118,15 +118,9 @@ RUN source /Miniforge/etc/profile.d/conda.sh \
118118
&& pip install --no-cache-dir ninja packaging psutil \
119119
&& pip install flash-attn==2.5.8 --no-build-isolation
120120

121-
# Add a new user "wildcodeuser"
122-
RUN adduser --disabled-password --gecos "" wildcodeuser
123-
124121
# Acquire benchmark code to local
125122
RUN git clone https://github.com/bigcode-project/code-eval.git /wildcode
126123

127-
RUN chown -R wildcodeuser:wildcodeuser /wildcode
128-
USER wildcodeuser
129-
130124
# Install Code-Eval and pre-load the dataset
131125
RUN source /Miniforge/etc/profile.d/conda.sh \
132126
&& source /Miniforge/etc/profile.d/mamba.sh \

‎Docker/Generate_Cuda12.Dockerfile‎

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -118,15 +118,9 @@ RUN source /Miniforge/etc/profile.d/conda.sh \
118118
&& pip install --no-cache-dir ninja packaging psutil \
119119
&& pip install flash-attn==2.5.8 --no-build-isolation
120120

121-
# Add a new user "wildcodeuser"
122-
RUN adduser --disabled-password --gecos "" wildcodeuser
123-
124121
# Acquire benchmark code to local
125122
RUN git clone https://github.com/bigcode-project/code-eval.git /wildcode
126123

127-
RUN chown -R wildcodeuser:wildcodeuser /wildcode
128-
USER wildcodeuser
129-
130124
# Install Code-Eval and pre-load the dataset
131125
RUN source /Miniforge/etc/profile.d/conda.sh \
132126
&& source /Miniforge/etc/profile.d/mamba.sh \

‎wildcode/generate.py‎

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ def codegen(
2222
n_samples=1,
2323
id_range=None,
2424
resume=True,
25-
subsample_size=None,
2625
):
2726
with Progress(
2827
TextColumn(f"{dataset} •" + "[progress.percentage]{task.percentage:>3.0f}%"),
@@ -33,11 +32,7 @@ def codegen(
3332
) as p:
3433
if dataset == "wildcodebench":
3534
from wildcode.data import get_wildcodebench, write_jsonl
36-
3735
dataset = get_wildcodebench()
38-
if subsample_size:
39-
if subsample_size < len(dataset):
40-
dataset = dataset[:subsample_size]
4136

4237
if model.is_direct_completion() and nl2code:
4338
raise Exception("Base model does not support direct completion for NL2Code tasks")
@@ -112,7 +107,6 @@ def main():
112107
parser.add_argument("--model", required=True, type=str)
113108
parser.add_argument("--dataset", required=True, type=str)
114109
parser.add_argument("--save_path", default=None, type=str)
115-
parser.add_argument("--subsample_size", default=None, type=int)
116110
parser.add_argument("--nl2code", action='store_true')
117111
parser.add_argument("--bs", default=1, type=int)
118112
parser.add_argument("--n_samples", default=1, type=int)
@@ -173,7 +167,6 @@ def main():
173167
n_samples=args.n_samples,
174168
resume=args.resume,
175169
id_range=args.id_range,
176-
subsample_size=args.subsample_size,
177170
)
178171

179172

0 commit comments

Comments
 (0)