karthikeya1212 commited on
Commit
057257b
·
verified ·
1 Parent(s): 07b5a8b

Update core/image_generator.py

Browse files
Files changed (1) hide show
  1. core/image_generator.py +1 -1
core/image_generator.py CHANGED
@@ -146,7 +146,7 @@ os.environ.update({
146
 
147
  # ---------------- MODEL CONFIG ----------------
148
  MODEL_REPO = "SG161222/RealVisXL_V4.0"
149
- MODEL_FILENAME = "realvisxlV40_v40LightningBakedvae.safetensors"
150
 
151
  # ---------------- MODEL DOWNLOAD ----------------
152
  def download_model() -> Path:
 
146
 
147
  # ---------------- MODEL CONFIG ----------------
148
  MODEL_REPO = "SG161222/RealVisXL_V4.0"
149
+ MODEL_FILENAME = "realvisxlV40_v40.safetensors"
150
 
151
  # ---------------- MODEL DOWNLOAD ----------------
152
  def download_model() -> Path: