Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -212,7 +212,7 @@ def get_ans(clip_emb, prompt):
|
|
212 |
|
213 |
|
214 |
device = 'cpu'
|
215 |
-
clip_model, preprocess = clip.load("ViT-L
|
216 |
tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3medium_based_on_gpt2')
|
217 |
prefix_length = 30
|
218 |
model_path = 'prefix_small_latest_gpt2_medium.pt'
|
|
|
212 |
|
213 |
|
214 |
device = 'cpu'
|
215 |
+
clip_model, preprocess = clip.load("ViT-L/14@336px", device=device, jit=False)
|
216 |
tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3medium_based_on_gpt2')
|
217 |
prefix_length = 30
|
218 |
model_path = 'prefix_small_latest_gpt2_medium.pt'
|