File tree Expand file tree Collapse file tree 2 files changed +6
-9
lines changed Expand file tree Collapse file tree 2 files changed +6
-9
lines changed Original file line number Diff line number Diff line change @@ -14,12 +14,9 @@ git clone https://github.com/pytorch-tpu/jetstream-pytorch.git
1414### 2. Run installation script:
1515
1616``` bash
17- source install_everything.sh
17+ sh install_everything.sh
1818```
1919
20- NOTE: the above script will export ` PYTHONPATH ` , so sourcing will make it
21- to take effect in the current shell
22-
2320
2421# Get weights
2522
Original file line number Diff line number Diff line change @@ -9,10 +9,10 @@ mkdir -p deps
99pushd deps
1010git clone https://github.com/google/JetStream.git
1111git clone https://github.com/pytorch/xla.git
12- pushd xla
12+ pushd xla/experimental/torch_xla2
1313git checkout $TORCHXLA_TAG
14- popd
14+ pip install .
1515popd # now at the folder of jetstream-pytorch
16-
17- export PYTHONPATH= $PYTHONPATH : $( pwd ) /deps/xla/experimental/torch_xla2: $( pwd ) /deps/JetStream: $( pwd )
18-
16+ pushd JetStream
17+ pip install .
18+ popd
You can’t perform that action at this time.
0 commit comments