Skip to content

Commit 6852632

Browse files
authored
Update README.md
1 parent fede128 commit 6852632

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ The tool is used for ranking LLM-generated Loop Invariants. We evaluate the inva
2121

2222

2323
## Requirement
24-
The requirements of this project are as follows. We built the tool based on python 3.9. We evaluated the project in [`PyTorch-1.13.1` (with cuda 11.6)](https://pytorch.org/get-started/previous-versions/#v1131) and [`Transformers-4.29.2`](https://github.com/huggingface/transformers/releases/tag/v4.29.2). Other versions of the libraries may need modification of the code, specially the [`trainer.py`](src/ranker/trainer.py) (and perhaps other places). To setup the appropriate library versions, run `pip install -r requirements.txt`.
24+
The requirements of this project are as follows. We built the tool based on python 3.9. We evaluated the project in [`PyTorch-1.13.1` (with cuda 11.6)](https://pytorch.org/get-started/previous-versions/#v1131) and [`Transformers-4.36.0`](https://github.com/huggingface/transformers/releases/tag/v4.36.0). Other versions of the libraries may need modification of the code, specially the [`trainer.py`](src/ranker/trainer.py) (and perhaps other places). To setup the appropriate library versions, run `pip install -r requirements.txt`.
2525

2626
## About the data
2727
We provide to version of the data. Follow the steps below to extract the datasets

0 commit comments

Comments
 (0)