Migrate to updated pd_models.get_models kw args

PAIR-code · Sep 23, 2024 · 821016e · 821016e
1 parent e4a6001
commit 821016e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lit_nlp/examples/gcp/model_server.py b/lit_nlp/examples/gcp/model_server.py
@@ -56,7 +56,7 @@ def _handler(app: wsgi_app.App, request, unused_environ):
       dl_runtime=dl_runtime,
       precision=precision,
       batch_size=batch_size,
-      sequence_length=sequence_length,
+      max_length=sequence_length,
   )
 
   gen_name = model_config[0].split(':')[0]