Upload dc.py

dc.py

@@ -46,6 +46,7 @@ ImageFile.LOAD_TRUNCATED_IMAGES = True
 # os.environ["PYTORCH_NO_CUDA_MEMORY_CACHING"] = "1"
 print(os.getenv("SPACES_ZERO_GPU"))
 
+## BEGIN MOD
 import gradio as gr
 import logging
 logging.getLogger("diffusers").setLevel(logging.ERROR)
@@ -66,9 +67,8 @@ from env import (
     DIRECTORY_MODELS, DIRECTORY_LORAS, DIRECTORY_VAES, DIRECTORY_EMBEDS,
     DIRECTORY_EMBEDS_SDXL, DIRECTORY_EMBEDS_POSITIVE_SDXL,
     LOAD_DIFFUSERS_FORMAT_MODEL, DOWNLOAD_MODEL_LIST, DOWNLOAD_LORA_LIST,
-    DOWNLOAD_VAE_LIST, download_embeds)
+    DOWNLOAD_VAE_LIST, DOWNLOAD_EMBEDS)
 
-## BEGIN MOD
 from modutils import (to_list, list_uniq, list_sub, get_model_id_list, get_tupled_embed_list,
                       get_tupled_model_list, get_lora_model_list, download_private_repo, download_things)
 
@@ -97,7 +97,7 @@ for url in [url.strip() for url in download_lora.split(',')]:
         download_things(DIRECTORY_LORAS, url, HF_TOKEN, CIVITAI_API_KEY)
 
 # Download Embeddings
-for url_embed in download_embeds:
+for url_embed in DOWNLOAD_EMBEDS:
     if not os.path.exists(f"./embedings/{url_embed.split('/')[-1]}"):
         download_things(DIRECTORY_EMBEDS, url_embed, HF_TOKEN, CIVITAI_API_KEY)
 
@@ -105,6 +105,7 @@ for url_embed in download_embeds:
 embed_list = get_model_list(DIRECTORY_EMBEDS)
 model_list = get_model_list(DIRECTORY_MODELS)
 model_list = load_diffusers_format_model + model_list
+
 ## BEGIN MOD
 lora_model_list = get_lora_model_list()
 vae_model_list = get_model_list(DIRECTORY_VAES)