Spaces:
Runtime error
Runtime error
Commit
·
a3bbfb3
1
Parent(s):
b264503
Upload 2 files
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
@@ -23,7 +23,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
23 |
feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
|
24 |
|
25 |
# load and configure tokenizer
|
26 |
-
tokenizer = AutoTokenizer.from_pretrained("facebook/opt-
|
27 |
tokenizer.pad_token = '!'
|
28 |
tokenizer.eos_token = '.'
|
29 |
|
|
|
23 |
feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
|
24 |
|
25 |
# load and configure tokenizer
|
26 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/opt-125m")
|
27 |
tokenizer.pad_token = '!'
|
28 |
tokenizer.eos_token = '.'
|
29 |
|
requirements.txt
CHANGED
@@ -12,7 +12,7 @@ torch==1.12.1
|
|
12 |
torchaudio==0.12.1
|
13 |
torchvision==0.13.1
|
14 |
tqdm==4.64.0
|
15 |
-
transformers
|
16 |
ftfy
|
17 |
regex
|
18 |
tqdm
|
|
|
12 |
torchaudio==0.12.1
|
13 |
torchvision==0.13.1
|
14 |
tqdm==4.64.0
|
15 |
+
transformers==4.21.1
|
16 |
ftfy
|
17 |
regex
|
18 |
tqdm
|