diff --git a/modules/expansion.py b/modules/expansion.py index ce40b569..a8f3d9c8 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -3,11 +3,11 @@ import torch import math import fcbh.model_management as model_management -from transformers.generation.logits_process import LogitsProcessorList from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed from modules.path import fooocus_expansion_path from fcbh.model_patcher import ModelPatcher + # limitation of np.random.seed(), called from transformers.set_seed() SEED_LIMIT_NUMPY = 2**32