diff --git a/modules/loaders.py b/modules/loaders.py index 062e4837..b8ae82d7 100644 --- a/modules/loaders.py +++ b/modules/loaders.py @@ -86,6 +86,7 @@ loaders_and_params = OrderedDict({ 'no_sdpa', 'exllamav2_info', 'model_draft', + 'draft_max', 'ctx_size_draft', 'speculative_decoding_accordion', ],