hmrizal commited on
Commit
93d9ba0
·
verified ·
1 Parent(s): 67219f4

undo removing getattr

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -213,7 +213,7 @@ def create_llm_pipeline(model_key):
213
  return llm
214
 
215
  # Create appropriate pipeline for HF models
216
- elif model_info.get("is_t5", False):
217
  print("Creating T5 pipeline")
218
  pipe = pipeline(
219
  "text2text-generation",
 
213
  return llm
214
 
215
  # Create appropriate pipeline for HF models
216
+ elif getattr(model_info, "is_t5", False):
217
  print("Creating T5 pipeline")
218
  pipe = pipeline(
219
  "text2text-generation",