Upload expansion.py

extras/expansion.py

@@ -14,8 +14,9 @@ from transformers.generation.logits_process import LogitsProcessorList
 from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
 # from modules.config import path_fooocus_expansion
 from ldm_patched.modules.model_patcher import ModelPatcher
+from huggingface_hub import hf_hub_download
 
-path_fooocus_expansion ="extras/fooocus_expansion"
+path_fooocus_expansion = hf_hub_download("aifeifei798/fooocus_expansion")
 # limitation of np.random.seed(), called from transformers.set_seed()
 SEED_LIMIT_NUMPY = 2**32
 neg_inf = - 8192.0