Spaces:
Running
Running
Commit
·
fe53098
1
Parent(s):
a1eae7e
Upload app.py
Browse files
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 |
|