fallenshock commited on
Commit
d00ce10
1 Parent(s): 07ae0ca
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ import huggingface_hub
12
 
13
 
14
  from FlowEdit_utils import FlowEditSD3, FlowEditFLUX
15
- SD3STRING = 'stabilityai/stable-diffusion-3-medium'
16
  FLUXSTRING = 'black-forest-labs/FLUX.1-dev'
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
18
  # device = "cpu"
@@ -90,7 +90,7 @@ def FlowEditRun(
90
  # except huggingface_hub.utils._errors.GatedRepoError:
91
  except huggingface_hub.errors.GatedRepoError:
92
  raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
93
- "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>"
94
  "model page</a> to get access.")
95
  # elif model_type == 'FLUX':
96
  # try:
 
12
 
13
 
14
  from FlowEdit_utils import FlowEditSD3, FlowEditFLUX
15
+ SD3STRING = 'stabilityai/stable-diffusion-3-medium-diffusers'
16
  FLUXSTRING = 'black-forest-labs/FLUX.1-dev'
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
18
  # device = "cpu"
 
90
  # except huggingface_hub.utils._errors.GatedRepoError:
91
  except huggingface_hub.errors.GatedRepoError:
92
  raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
93
+ "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers'>"
94
  "model page</a> to get access.")
95
  # elif model_type == 'FLUX':
96
  # try: