Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +9 -0
requirements.txt
CHANGED
@@ -11,4 +11,13 @@ pip install gradio
|
|
11 |
pip install torch==2.0.1
|
12 |
pip install accelerate
|
13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
|
|
|
11 |
pip install torch==2.0.1
|
12 |
pip install accelerate
|
13 |
|
14 |
+
import torch
|
15 |
+
import transformers
|
16 |
+
import numpy as np
|
17 |
+
import torch.nn.functional as F
|
18 |
+
from accelerate import Accelerator
|
19 |
+
from torch.utils.data import Dataset
|
20 |
+
from transformers import AutoConfig, AutoModelForCausalLM, GPT2Tokenizer
|
21 |
+
from transformers import GPT2Tokenizer
|
22 |
+
|
23 |
|