Update app.py
Browse files
app.py
CHANGED
|
@@ -144,9 +144,10 @@ def run_interpretation(raw_interpretation_prompt, max_new_tokens, do_sample,
|
|
| 144 |
|
| 145 |
|
| 146 |
## main
|
| 147 |
-
global_state = reset_model(model_name, return_state=True)
|
| 148 |
torch.set_grad_enabled(False)
|
|
|
|
| 149 |
model_name = 'LLAMA2-7B'
|
|
|
|
| 150 |
original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
|
| 151 |
tokens_container = []
|
| 152 |
for i in range(MAX_PROMPT_TOKENS):
|
|
|
|
| 144 |
|
| 145 |
|
| 146 |
## main
|
|
|
|
| 147 |
torch.set_grad_enabled(False)
|
| 148 |
+
|
| 149 |
model_name = 'LLAMA2-7B'
|
| 150 |
+
global_state = reset_model(model_name, return_state=True)
|
| 151 |
original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
|
| 152 |
tokens_container = []
|
| 153 |
for i in range(MAX_PROMPT_TOKENS):
|