Spaces:
Runtime error
Runtime error
Tao Wu commited on
Commit ·
9b1fd32
1
Parent(s): 4486afa
init
Browse files- app/embedding_setup.py +1 -1
app/embedding_setup.py
CHANGED
|
@@ -173,7 +173,7 @@ def generate_exp(
|
|
| 173 |
):
|
| 174 |
|
| 175 |
|
| 176 |
-
resp = lorax_client.generate(prompt,adapter_id=
|
| 177 |
|
| 178 |
return resp.generated_text
|
| 179 |
|
|
|
|
| 173 |
):
|
| 174 |
|
| 175 |
|
| 176 |
+
resp = lorax_client.generate(prompt,adapter_id=EXP_LORA_MODEL, adapter_source='hub', max_new_tokens=max_new_tokens)
|
| 177 |
|
| 178 |
return resp.generated_text
|
| 179 |
|