dar-tau commited on
Commit
dafa3d0
·
verified ·
1 Parent(s): 293df90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -158,7 +158,7 @@ torch.set_grad_enabled(False)
158
  global_state = GlobalState()
159
 
160
  model_name = 'LLAMA2-7B'
161
- reset_model(model_name, with_extra_components=False)
162
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
163
  tokens_container = []
164
 
 
158
  global_state = GlobalState()
159
 
160
  model_name = 'LLAMA2-7B'
161
+ reset_model(model_name, with_extra_components=False, reset_sentence_transformer=True)
162
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
163
  tokens_container = []
164