Spaces:
Sleeping
Sleeping
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 |
|