Spaces:
Running
Running
undo removing getattr
Browse files
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
|
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",
|