Update app.py
Browse files
app.py
CHANGED
@@ -151,7 +151,7 @@ torch.set_grad_enabled(False)
|
|
151 |
global_state = GlobalState()
|
152 |
|
153 |
model_name = 'LLAMA2-7B'
|
154 |
-
reset_model(model_name,
|
155 |
original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
|
156 |
tokens_container = []
|
157 |
|
|
|
151 |
global_state = GlobalState()
|
152 |
|
153 |
model_name = 'LLAMA2-7B'
|
154 |
+
reset_model(model_name, return_extra_components=False)
|
155 |
original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
|
156 |
tokens_container = []
|
157 |
|