Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,8 @@ An extensible, convenient, and efficient toolbox for finetuning large machine le

### Setup

Our package has been fully tested on Linux OS (Ubuntu 20.04). Other OS platforms (MacOS, Windows) are not fully tested.
You may encounter some unexpected errors. You may try it first on a Linux machine or use Google Colab to experience it.
Our package has been tested on Linux OS (Ubuntu 20.04). Other OS platforms (MacOS, Windows) are not fully tested, where you may encounter unexpected errors. If you are using LMFlow for the first time, we recommend you to try on a Linux machine or Google Colab.

CUDA versions 10.3-11.7 are supported in versions `v0.0.5` or older. For CUDA versions greater than 11.7, one can use our stable branch `>= v0.0.6`.

```bash
Expand All @@ -102,7 +102,8 @@ Please refer to our [doc](https://optimalscale.github.io/LMFlow/examples/DATASET

### Finetuning (Full)
Full training updates all the parameters to finetune a language model.
Here is an example to finetune a GPT-2 base model. (We encounter a data server problem recently. To avoid errors, please use the latest [`download.sh`](https://github.com/OptimalScale/LMFlow/blob/main/data/download.sh) script in main branch to download the dataset)
Here is an example to finetune a GPT-2 base model.

```sh
cd data && ./download.sh alpaca && cd -

Expand All @@ -113,8 +114,7 @@ cd data && ./download.sh alpaca && cd -
```

### Finetuning (LISA)
[LISA](https://arxiv.org/abs/2403.17919) is a memory-efficient finetuning algorithm that allows tradeoff between memory and the number of randomly unfreezed layers. This script currently is only tested in single gpus. Please stay tuned for our latest updates :smile: (We encounter a data server problem recently. To avoid errors, please use the latest [`download.sh`](https://github.com/OptimalScale/LMFlow/blob/main/data/download.sh) script in main branch to download the dataset)

[LISA](https://arxiv.org/abs/2403.17919) is a memory-efficient finetuning algorithm that allows tradeoff between memory and the number of randomly unfreezed layers. This script currently is only tested in single gpus. Please stay tuned for our latest updates :smile:
```sh
cd data && ./download.sh alpaca && cd -

Expand Down
2 changes: 1 addition & 1 deletion data/download.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#!/bin/bash

function main() {
public_server="http://107.23.182.175:5000"
public_server="http://lmflow.org:5000"
if [ $# -lt 1 -o "$1" = "-h" -o "$1" = "--help" ]; then
echo "Usage: bash $(basename $0) dataset_name"
echo "Example: bash $(basename $0) MedMCQA"
Expand Down