Cagliostro commited on
Commit
8c4042b
1 Parent(s): 218ef1d

fix model url

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
31
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
32
  OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
33
 
34
- MODEL = os.getenv("MODEL", "cagliostrolab/animagine-xl-3.1")
35
 
36
  torch.backends.cudnn.deterministic = True
37
  torch.backends.cudnn.benchmark = False
@@ -78,7 +78,7 @@ def generate(
78
  style_selector: str = "(None)",
79
  quality_selector: str = "Standard v3.1",
80
  use_upscaler: bool = False,
81
- upscaler_strength: float = 0.5,
82
  upscale_by: float = 1.5,
83
  add_quality_tags: bool = True,
84
  progress=gr.Progress(track_tqdm=True),
 
31
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
32
  OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
33
 
34
+ MODEL = os.getenv("MODEL", "https://huggingface.co/cagliostrolab/animagine-xl-3.1/blob/main/animagine-xl-3.1.safetensors")
35
 
36
  torch.backends.cudnn.deterministic = True
37
  torch.backends.cudnn.benchmark = False
 
78
  style_selector: str = "(None)",
79
  quality_selector: str = "Standard v3.1",
80
  use_upscaler: bool = False,
81
+ upscaler_strength: float = 0.55,
82
  upscale_by: float = 1.5,
83
  add_quality_tags: bool = True,
84
  progress=gr.Progress(track_tqdm=True),