c83908758 commited on
Commit
98400a2
·
verified ·
1 Parent(s): 062c4e3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -62,7 +62,6 @@ def load_pipe():
62
  pipe = AutoPipelineForText2Image.from_pretrained(
63
  MODEL_ID,
64
  torch_dtype=_dtype,
65
- use_safetensors=True,
66
  )
67
  if hasattr(pipe, "enable_attention_slicing"):
68
  pipe.enable_attention_slicing()
@@ -201,4 +200,4 @@ with gr.Blocks(title="wode") as demo:
201
  unload.click(fn=unload_pipe, outputs=[])
202
 
203
  if __name__ == "__main__":
204
- demo.queue(max_size=4).launch()
 
62
  pipe = AutoPipelineForText2Image.from_pretrained(
63
  MODEL_ID,
64
  torch_dtype=_dtype,
 
65
  )
66
  if hasattr(pipe, "enable_attention_slicing"):
67
  pipe.enable_attention_slicing()
 
200
  unload.click(fn=unload_pipe, outputs=[])
201
 
202
  if __name__ == "__main__":
203
+ demo.queue(max_size=4).launch(show_error=True)