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