Spaces:
Build error
Build error
Fix queue comment
Browse files- app.py +2 -2
- fromage/models.py +1 -1
app.py
CHANGED
|
@@ -212,5 +212,5 @@ with gr.Blocks(css=css) as demo:
|
|
| 212 |
save_button.click(None, [], [], _js=save_js)
|
| 213 |
|
| 214 |
|
| 215 |
-
|
| 216 |
-
demo.launch(debug=True, server_name="
|
|
|
|
| 212 |
save_button.click(None, [], [], _js=save_js)
|
| 213 |
|
| 214 |
|
| 215 |
+
demo.queue(concurrency_count=1, api_open=False, max_size=16)
|
| 216 |
+
demo.launch(debug=True, server_name="0.0.0.0")
|
fromage/models.py
CHANGED
|
@@ -640,7 +640,7 @@ def load_fromage(embeddings_dir: str, model_args_path: str, model_ckpt_path: str
|
|
| 640 |
assert len(ret_token_idx) == 1, ret_token_idx
|
| 641 |
model_kwargs['retrieval_token_idx'] = ret_token_idx[0]
|
| 642 |
|
| 643 |
-
debug =
|
| 644 |
if debug:
|
| 645 |
model_kwargs['opt_version'] = 'facebook/opt-125m'
|
| 646 |
model_kwargs['visual_encoder'] = 'openai/clip-vit-base-patch32'
|
|
|
|
| 640 |
assert len(ret_token_idx) == 1, ret_token_idx
|
| 641 |
model_kwargs['retrieval_token_idx'] = ret_token_idx[0]
|
| 642 |
|
| 643 |
+
debug = False
|
| 644 |
if debug:
|
| 645 |
model_kwargs['opt_version'] = 'facebook/opt-125m'
|
| 646 |
model_kwargs['visual_encoder'] = 'openai/clip-vit-base-patch32'
|