Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Megatron-LM
Commits
e718810e
Commit
e718810e
authored
Aug 27, 2021
by
rprenger
Browse files
Removing NVIDIA specific code and fixing some whitespace
parent
448cb299
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
2 additions
and
31 deletions
+2
-31
examples/run_text_generation_server_345M.sh
examples/run_text_generation_server_345M.sh
+1
-1
examples/run_text_generation_server_345M_8_tensor_parallel.sh
...ples/run_text_generation_server_345M_8_tensor_parallel.sh
+1
-1
run_api_server_8.3B.sh
run_api_server_8.3B.sh
+0
-29
No files found.
examples/run_text_generation_server_345M.sh
View file @
e718810e
...
...
@@ -29,4 +29,4 @@ python -m torch.distributed.launch $DISTRIBUTED_ARGS tools/run_text_generation_s
--vocab-file
$VOCAB_FILE
/
--merge-file
$MERGE_FILE
/
--top_p
0.9 /
--seed
42
--seed
42
examples/run_text_generation_server_345M_8_tensor_parallel.sh
View file @
e718810e
...
...
@@ -29,4 +29,4 @@ python -m torch.distributed.launch $DISTRIBUTED_ARGS tools/run_text_generation_s
--vocab-file
$VOCAB_FILE
/
--merge-file
$MERGE_FILE
/
--top_p
0.9 /
--seed
42
--seed
42
run_api_server_8.3B.sh
deleted
100755 → 0
View file @
448cb299
#!/bin/bash
WORLD_SIZE
=
8
DISTRIBUTED_ARGS
=
"--nproc_per_node
$WORLD_SIZE
\
--nnodes 1
\
--node_rank 0
\
--master_addr localhost
\
--master_port 6000"
CHECKPOINT
=
"/home/universal-lm-data.cosmos549/chkpts/gpt2/8.3B_no_rng"
DATA_PATH
=
"/home/universal-lm-data.cosmos549/scratch/mshoeybi/data/gpt2"
VOCAB_FILE
=
"
${
DATA_PATH
}
/bpe/gpt2-vocab.json"
MERGE_FILE
=
"
${
DATA_PATH
}
/bpe/gpt2-merges.txt"
python
-m
torch.distributed.launch
$DISTRIBUTED_ARGS
tools/run_api_server.py
\
--tensor-model-parallel-size
8
\
--pipeline-model-parallel-size
1
\
--num-layers
72
\
--hidden-size
3072
\
--load
$CHECKPOINT
\
--num-attention-heads
24
\
--max-position-embeddings
1024
\
--tokenizer-type
GPT2BPETokenizer
\
--fp16
\
--micro-batch-size
1
\
--seq-length
1024
\
--out-seq-length
1024
\
--temperature
1.0
\
--vocab-file
$VOCAB_FILE
\
--merge-file
$MERGE_FILE
\
--top_p
0.9
\
--seed
42
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment