Spaces:
Runtime error
Runtime error
ffreemt
commited on
Commit
·
c2783a5
1
Parent(s):
4410eba
Update reset=True
Browse files
app.py
CHANGED
@@ -142,9 +142,13 @@ def generate(
|
|
142 |
"""Run model inference, will return a Generator if streaming is true."""
|
143 |
# _ = prompt_template.format(question=question)
|
144 |
# print(_)
|
|
|
|
|
|
|
145 |
return llm(
|
146 |
question,
|
147 |
-
**asdict(generation_config),
|
|
|
148 |
)
|
149 |
|
150 |
|
@@ -357,11 +361,7 @@ with gr.Blocks(
|
|
357 |
f"""<h5><center>{Path(model_loc).name}</center></h4>
|
358 |
Most examples are meant for another model.
|
359 |
You probably should try to test
|
360 |
-
some related prompts.
|
361 |
-
|
362 |
-
Question: {{question}}
|
363 |
-
|
364 |
-
Answer: Let's work this out in a step by step way to be sure we have the right answer.""",
|
365 |
elem_classes="xsmall",
|
366 |
)
|
367 |
|
|
|
142 |
"""Run model inference, will return a Generator if streaming is true."""
|
143 |
# _ = prompt_template.format(question=question)
|
144 |
# print(_)
|
145 |
+
|
146 |
+
config = GenerationConfig(reset=True) # rid of OOM?
|
147 |
+
|
148 |
return llm(
|
149 |
question,
|
150 |
+
# **asdict(generation_config),
|
151 |
+
**asdict(config),
|
152 |
)
|
153 |
|
154 |
|
|
|
361 |
f"""<h5><center>{Path(model_loc).name}</center></h4>
|
362 |
Most examples are meant for another model.
|
363 |
You probably should try to test
|
364 |
+
some related prompts.""",
|
|
|
|
|
|
|
|
|
365 |
elem_classes="xsmall",
|
366 |
)
|
367 |
|