Fix issue when load AutoModelForSeq2SeqLM model (#370)

5fde8d99 · CL-Shang · GitHub · 62f91f78 · 5fde8d99
Unverified Commit 5fde8d99 authored May 26, 2023 by CL-Shang Committed by GitHub May 26, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

server/text_generation_server/models/__init__.py server/text_generation_server/models/__init__.py +1 -1

No files found.
--- a/server/text_generation_server/models/__init__.py
+++ b/server/text_generation_server/models/__init__.py
@@ -266,7 +266,7 @@ def get_model(
                quantize=quantize,
                trust_remote_code=trust_remote_code,
            )
-        if "AutoModelForSeq2SeqLM" in auto_map.keys:
+        if "AutoModelForSeq2SeqLM" in auto_map.keys():
            return Seq2SeqLM(
                model_id,
                revision,