Update app.py
Browse files
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 |
|