prithivMLmods commited on
Commit
29efcc3
·
verified ·
1 Parent(s): a6ae91e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ gr.Markdown()
13
  DESCRIPTIONx="""ㅤㅤㅤ## Midjourney XL"""
14
 
15
  # Use environment variables for flexibility
16
- MODEL_ID = os.getenv("MODEL_ID", "SG161222/RealVisXL_V4.0_Lightning")
17
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
18
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
19
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
13
  DESCRIPTIONx="""ㅤㅤㅤ## Midjourney XL"""
14
 
15
  # Use environment variables for flexibility
16
+ MODEL_ID = os.getenv("MODEL_REPO")
17
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
18
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
19
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"