Sort some imports

2023-06-25 01:44:36 -03:00 · 2023-06-25 01:44:36 -03:00 · f0fcd1f697
commit f0fcd1f697
parent 365b672531
8 changed files with 60 additions and 33 deletions
--- a/modules/sampler_hijack.py
+++ b/modules/sampler_hijack.py
@ -3,9 +3,11 @@ import math
 import torch
 import transformers
 from transformers import LogitsWarper
-from transformers.generation.logits_process import (LogitNormalization,
-                                                    LogitsProcessorList,
-                                                    TemperatureLogitsWarper)
+from transformers.generation.logits_process import (
+    LogitNormalization,
+    LogitsProcessorList,
+    TemperatureLogitsWarper
+)


 class TailFreeLogitsWarper(LogitsWarper):