From 203783a58ca4fd8ff7b109f0eeb6826948a124d6 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 27 Sep 2024 15:02:36 -0700 Subject: [PATCH] Remove debug statements --- modules/sampler_hijack.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/modules/sampler_hijack.py b/modules/sampler_hijack.py index c2e16fa9..9e865bb3 100644 --- a/modules/sampler_hijack.py +++ b/modules/sampler_hijack.py @@ -546,10 +546,8 @@ def get_logits_processor_patch(self, **kwargs): # Return a large value if class name is not mapped or if the mapped nickname is not in priority if class_name not in class_name_to_nickname or class_name_to_nickname[class_name] not in sampler_priority: - print("======>", class_name, "inf") return float('inf') - print("======>", class_name, sampler_priority.index(class_name_to_nickname[class_name])) # Return the index of the nickname in the priority list for sorting return sampler_priority.index(class_name_to_nickname[class_name])