rajux75 commited on
Commit
204e1ea
·
verified ·
1 Parent(s): f5a4acf

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -0
config.py CHANGED
@@ -8,6 +8,7 @@ load_dotenv() # Load variables from .env file for local development
8
  # Model Configuration
9
  TEXT_MODEL_NAME = os.getenv("TEXT_MODEL_NAME", "google/flan-t5-base")
10
  IMAGE_MODEL_NAME = os.getenv("IMAGE_MODEL_NAME", "runwayml/stable-diffusion-v1-5")
 
11
  VIDEO_MODEL_NAME = os.getenv("VIDEO_MODEL_NAME", "zeroscope_v2_576w")
12
 
13
  # API Security
 
8
  # Model Configuration
9
  TEXT_MODEL_NAME = os.getenv("TEXT_MODEL_NAME", "google/flan-t5-base")
10
  IMAGE_MODEL_NAME = os.getenv("IMAGE_MODEL_NAME", "runwayml/stable-diffusion-v1-5")
11
+ IMAGE_LCM_LORA_NAME = "latent-consistency/lcm-lora-sdv1-5" # Example LCM LoRA
12
  VIDEO_MODEL_NAME = os.getenv("VIDEO_MODEL_NAME", "zeroscope_v2_576w")
13
 
14
  # API Security