jykoh commited on
Commit
2feda0d
1 Parent(s): bf9abdd

Fix deployment of debug model

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. fromage/models.py +1 -1
app.py CHANGED
@@ -208,6 +208,6 @@ with gr.Blocks(css=css) as demo:
208
  save_button.click(None, [], [], _js=save_js)
209
 
210
 
211
- # demo.queue(concurrency_count=1, api_open=False, max_size=16)
212
- # demo.launch(debug=True, server_name="0.0.0.0")
213
- demo.launch(debug=True, server_name="127.0.0.1")
 
208
  save_button.click(None, [], [], _js=save_js)
209
 
210
 
211
+ demo.queue(concurrency_count=1, api_open=False, max_size=16)
212
+ demo.launch(debug=True, server_name="0.0.0.0")
213
+ # demo.launch(debug=True, server_name="127.0.0.1")
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 = True
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'