freQuensy23 commited on
Commit
c1e500d
1 Parent(s): a66220d

Update generators.py

Browse files
Files changed (1) hide show
  1. generators.py +1 -1
generators.py CHANGED
@@ -41,7 +41,7 @@ async def generate_mistral_7bvo1(system_input, user_input):
41
  async def generate_t5(system_input, user_input):
42
  output = await query_llm({
43
  "inputs": (inputs := f"{system_input}\n{user_input}"),
44
- }, "google/flan-t5-large")
45
  try:
46
  yield output[0]["generated_text"]
47
  except (IndexError, KeyError):
 
41
  async def generate_t5(system_input, user_input):
42
  output = await query_llm({
43
  "inputs": (inputs := f"{system_input}\n{user_input}"),
44
+ }, "google/flan-t5-xxl")
45
  try:
46
  yield output[0]["generated_text"]
47
  except (IndexError, KeyError):