Fix temperature_last when temperature not in sampler priority (#6439)

oobabooga · Oct 9, 2024 · 03a2e70 · 03a2e70
1 parent 9d8b1c5
commit 03a2e70
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/modules/sampler_hijack.py b/modules/sampler_hijack.py
@@ -571,11 +571,10 @@ def get_logits_processor_patch(self, **kwargs):
     if generation_config.temperature_last:
         for param_name in ['temperature', 'dynamic_temperature', 'quadratic_sampling']:
             if param_name in sampler_priority:
-                if param_name in sampler_priority:
-                    index = sampler_priority.index(param_name)
-                    sampler_priority.append(sampler_priority.pop(index))
-                else:
-                    sampler_priority.append(param_name)
+                index = sampler_priority.index(param_name)
+                sampler_priority.append(sampler_priority.pop(index))
+            else:
+                sampler_priority.append(param_name)
 
     class_name_to_nickname = {
         'DynamicTemperatureLogitsWarper': 'dynamic_temperature',