Update handler.py
Browse files- handler.py +1 -3
handler.py
CHANGED
|
@@ -175,9 +175,7 @@ class EndpointHandler:
|
|
| 175 |
|
| 176 |
"generated_tokens": int((output_ids.shape[-1] - inputs["input_ids"].shape[-1])),
|
| 177 |
|
| 178 |
-
},
|
| 179 |
-
|
| 180 |
-
}
|
| 181 |
|
| 182 |
|
| 183 |
# Local quick test
|
|
|
|
| 175 |
|
| 176 |
"generated_tokens": int((output_ids.shape[-1] - inputs["input_ids"].shape[-1])),
|
| 177 |
|
| 178 |
+
},}
|
|
|
|
|
|
|
| 179 |
|
| 180 |
|
| 181 |
# Local quick test
|