patrickvonplaten commited on
Commit
f5c87bd
1 Parent(s): 7f4d207

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -0
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - opt_metasq
4
+ ---
5
+
6
+ # This repo let's you run the following checkpoint using facebookresearch/metaseq.
7
+
8
+ Do the following:
9
+
10
+ ## 1. Install PyTorch
11
+ ```
12
+ pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
13
+ ```
14
+
15
+ ## 2. Install Megatron
16
+ ```
17
+ git clone https://github.com/patrickvonplaten/Megatron-LM.git
18
+ cd Megatron-LM
19
+ pip3 install six regex
20
+ pip3 install -e .
21
+ ```
22
+ ## 3. Install fairscale
23
+ ```
24
+ git clone https://github.com/facebookresearch/fairscale.git
25
+ cd fairscale
26
+ git checkout prefetch_fsdp_params_simple
27
+ pip3 install -e .
28
+ ```
29
+ ## 4. Install metaseq
30
+ ```
31
+ git clone https://github.com/patrickvonplaten/metaseq.git
32
+ cd metaseq
33
+ pip3 install -e .
34
+ ```
35
+
36
+ ## 5. Clone this repo (click top right on "How to clone")
37
+
38
+ ## 6. Run the following:
39
+
40
+ ```bash
41
+ cd <path/to/cloned/repo>
42
+ bash run.sh
43
+ ```