camilaaeromoca commited on
Commit
82e4184
·
verified ·
1 Parent(s): 13defca

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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