Update app.py
Browse files
app.py
CHANGED
@@ -73,6 +73,7 @@ def generate_interpretation_gpu(interpret_prompt, *args, **kwargs):
|
|
73 |
return interpret_prompt.generate(*args, **kwargs)
|
74 |
|
75 |
|
|
|
76 |
def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens, do_sample,
|
77 |
temperature, top_k, top_p, repetition_penalty, length_penalty, use_gpu, i,
|
78 |
num_beams=1):
|
|
|
73 |
return interpret_prompt.generate(*args, **kwargs)
|
74 |
|
75 |
|
76 |
+
@spaces.GPU
|
77 |
def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens, do_sample,
|
78 |
temperature, top_k, top_p, repetition_penalty, length_penalty, use_gpu, i,
|
79 |
num_beams=1):
|