collin commited on
Commit
b582e82
·
1 Parent(s): 703ba76

update flux 1.1 pro detection

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,7 +90,7 @@ try:
90
  format="%.2f"
91
  )
92
 
93
- if model_version.startswith("pro"):
94
  guidance = st.slider(
95
  "Guidance - How closely the model follows your prompt, 2-5, default is 3",
96
  min_value=2.0,
@@ -125,7 +125,7 @@ try:
125
  step=1
126
  )
127
 
128
- if not model_version.startswith("pro"):
129
  safety_checker = "On"
130
  # safety_checker = st.radio(
131
  # "Safety Checker - Turn on model NSFW checking",
 
90
  format="%.2f"
91
  )
92
 
93
+ if model_version.startswith("pro") or model_version.startswith("1"):
94
  guidance = st.slider(
95
  "Guidance - How closely the model follows your prompt, 2-5, default is 3",
96
  min_value=2.0,
 
125
  step=1
126
  )
127
 
128
+ if not model_version.startswith("pro") and not model_version.startswith("1"):
129
  safety_checker = "On"
130
  # safety_checker = st.radio(
131
  # "Safety Checker - Turn on model NSFW checking",