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'
|