Files

 main

/

environment.yml

Blame

Blame

Latest commit

 

History

History

23 lines (23 loc) · 405 Bytes

 main

/

environment.yml

Top

File metadata and controls

• Code
• Blame

23 lines (23 loc) · 405 Bytes

Raw

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

name: gbai_cuda_lm

channels:

- conda-forge

- pytorch

dependencies:

- python=3.9

- pip=24.0

- pytorch::pytorch>=2.0

- pip:

- sentencepiece

- huggingface-hub

- transformers>=4.38.0

- accelerate

- colorama

- datasets

- lm-eval==0.3.0

- termcolor

- pillow

- requests

- prompt-toolkit

- rich

- optimum

- auto-gptq