Tao Wu commited on
Commit
9b1fd32
1 Parent(s): 4486afa
Files changed (1) hide show
  1. 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="wt3639/Llama-3-8B-Instruct_RecExp_lora", adapter_source='hub', max_new_tokens=max_new_tokens)
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