Spaces:

Ravinandan
/

test_space

Sleeping

Ravinandan commited on Apr 4

Commit

215690d

verified ·

1 Parent(s): 7d91294

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ pipeline = AutoPipelineForImage2Image.from_pretrained(
     use_safetensors=True
 )
 pipeline.enable_model_cpu_offload()
-pipeline.enable_xformers_memory_efficient_attention()
 # Upload an image
 uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])

     use_safetensors=True
 )
 pipeline.enable_model_cpu_offload()
 # Upload an image
 uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])