AchyuthGamer commited on
Commit
25ad7ae
1 Parent(s): e66bafa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,14 +11,14 @@ from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
11
  from previewer.modules import Previewer
12
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
13
 
14
- DESCRIPTION = "# Würstchen"
15
- DESCRIPTION += "\n<p style=\"text-align: center\"><a href='https://huggingface.co/spaces/AchyuthGamer/ImMagician-Image-Generator' target='_blank'>ImMagician</a> is a new fast and efficient high resolution text-to-image architecture and model</p>"
16
  if not torch.cuda.is_available():
17
- DESCRIPTION += "\n<p>ImMagician Image Generator 🪄</p>"
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
21
- MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
22
  USE_TORCH_COMPILE = False
23
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
24
  PREVIEW_IMAGES = True
 
11
  from previewer.modules import Previewer
12
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
13
 
14
+ DESCRIPTION = "ImMagician🪄"
15
+ DESCRIPTION += "\n<p style=\"text-align: center\"><a href='https://huggingface.co/warp-ai/wuerstchen' target='_blank'>ImMagician🪄</a> is a new fast and efficient high resolution text-to-image architecture and model</p>"
16
  if not torch.cuda.is_available():
17
+ DESCRIPTION += "\n<p>Running on CPU 🥶</p>"
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
21
+ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
22
  USE_TORCH_COMPILE = False
23
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
24
  PREVIEW_IMAGES = True