Skip to content

Commit 26dbbc7

Browse files
committed
unbreak the tests oops
1 parent 84a2aa0 commit 26dbbc7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test_gpt2.cu

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,16 @@ float* float_cpu_malloc_and_point_parameters(FloatParameterTensors* params, size
8585
int main(int argc, char *argv[]) {
8686
common_start(false, true);
8787

88+
// set the right paths
89+
#if defined(ENABLE_BF16)
90+
const char* load_filename = "gpt2_124M_bf16.bin";
91+
#else
92+
const char* load_filename = "gpt2_124M.bin";
93+
#endif
94+
8895
// build the GPT-2 model from a checkpoint
8996
GPT2 model;
97+
9098
gpt2_build_from_checkpoint(&model, load_filename);
9199
size_t V = model.config.vocab_size;
92100
size_t Vp = model.config.padded_vocab_size;

0 commit comments

Comments
 (0)