jykoh commited on
Commit
ffe60c6
1 Parent(s): 1ec724c

Set debug to false

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