Upload 2 files

app.py

@@ -6,7 +6,7 @@ import warnings
 warnings.filterwarnings('ignore')
 
 # Constants
-MODEL_ID = "runwayml/stable-diffusion-v1-5"  # Using SD 1.5 for better compatibility
+MODEL_ID = "CompVis/stable-diffusion-v1-4"  # Using SD 1.4 for better compatibility
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float32  # Using float32 for better compatibility
 
@@ -16,18 +16,19 @@ class FooocusGenerator:
         self.load_models()
 
     def load_models(self):
-        # Load base model
-        scheduler = DDIMScheduler.from_pretrained(MODEL_ID, subfolder="scheduler")
+        # Load base model with minimal memory usage
         self.pipe = StableDiffusionPipeline.from_pretrained(
             MODEL_ID,
-            scheduler=scheduler,
             torch_dtype=DTYPE,
-            safety_checker=None
+            safety_checker=None,
+            requires_safety_checker=False
         )
         
         if DEVICE == "cuda":
             self.pipe.enable_attention_slicing()
-        self.pipe = self.pipe.to(DEVICE)
+            self.pipe.enable_sequential_cpu_offload()
+        else:
+            self.pipe = self.pipe.to(DEVICE)
 
     def generate_image(
         self,
@@ -100,7 +101,7 @@ class FooocusGenerator:
 generator = FooocusGenerator()
 
 # Create the Gradio interface
-with gr.Blocks(title="Fooocus Web", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Fooocus Web", theme=gr.themes.Default()) as demo:
     gr.Markdown("# 🎨 Fooocus Web")
     gr.Markdown("Generate high-quality images with advanced controls")
     
@@ -197,4 +198,4 @@ with gr.Blocks(title="Fooocus Web", theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()

requirements.txt

@@ -3,13 +3,13 @@ torchvision==0.15.2
 transformers==4.30.2
 accelerate==0.21.0
 safetensors==0.3.1
-gradio==4.19.2
+gradio==3.50.2
 diffusers==0.19.3
 opencv-python==4.8.0.74
 einops==0.6.1
 pytorch-lightning==2.0.2
 omegaconf==2.3.0
-huggingface-hub==0.16.4
+huggingface-hub==0.15.1
 xformers==0.0.20
 triton==2.0.0
 compel==2.0.1