fix

flexflow · Mar 30, 2024 · e0a6e4f · e0a6e4f
1 parent 5bd7123
commit e0a6e4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/flexflow/serve/serve.py b/python/flexflow/serve/serve.py
@@ -375,7 +375,7 @@ def compile(
         self.rm.set_max_spec_tree_token_num(
             self.model_configs.max_spec_tree_token_num
             if "max_spec_tree_token_num"
-            in self.model_configs.max_spec_tree_token_num.__dict__
+            in self.model_configs.__dict__
             else 20
         )