jdavis commited on
Commit
0e1643e
·
verified ·
1 Parent(s): 011b800

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ def load_model():
103
  "black-forest-labs/FLUX.1-Fill-dev",
104
  token=token,
105
  torch_dtype=torch.bfloat16,
106
- tokenizer_config={"add_prefix_space": False}
107
  )
108
  st.success("Model loaded successfully!")
109
  return model.to(device)
 
103
  "black-forest-labs/FLUX.1-Fill-dev",
104
  token=token,
105
  torch_dtype=torch.bfloat16,
106
+ revision="main"
107
  )
108
  st.success("Model loaded successfully!")
109
  return model.to(device)