Removed gpu offloading

app.py

@@ -37,7 +37,7 @@ def update_default_concepts(prompt):
 
     return gr.update(value=default_concepts.get(prompt, []))
 
-pipeline = ConceptAttentionFluxPipeline(model_name="flux-schnell", device="cuda", offload_model=True)
+pipeline = ConceptAttentionFluxPipeline(model_name="flux-schnell", device="cuda") #, offload_model=True)
 
 def convert_pil_to_bytes(img):
     img = img.resize((IMG_SIZE, IMG_SIZE), resample=Image.NEAREST)
@@ -271,7 +271,7 @@ with gr.Blocks(
         )
 
 if __name__ == "__main__":
-    demo.launch(max_threads=1)
+    demo.launch()
     #     share=True,
     #     server_name="0.0.0.0",
     #     inbrowser=True,