Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
cbf3592
1
Parent(s):
c4fbba7
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -255,6 +255,7 @@ class Summarization:
|
|
255 |
model_name (str, optional): exact model architecture name, "t5-base" or "t5-large". Defaults to "t5-base".
|
256 |
:param model_type:
|
257 |
"""
|
|
|
258 |
|
259 |
def train(
|
260 |
self,
|
|
|
255 |
model_name (str, optional): exact model architecture name, "t5-base" or "t5-large". Defaults to "t5-base".
|
256 |
:param model_type:
|
257 |
"""
|
258 |
+
if model_type == "t5":
|
259 |
|
260 |
def train(
|
261 |
self,
|