Rollback Working except Avatar

app.py

@@ -11,6 +11,7 @@ import numpy as np
 from PIL import Image, ImageEnhance, ImageOps
 from rembg import remove
 from diffusers import StableDiffusionImg2ImgPipeline
+from accelerate import Accelerator
 import io
 
 
@@ -24,6 +25,8 @@ from gfpgan import GFPGANer
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from realesrgan import RealESRGANer
 
+accelerator = Accelerator()
+
 # ------------------------------------------
 # Model Loading (Outside GPU decorator)
 # ------------------------------------------
@@ -66,8 +69,11 @@ def load_models():
         sd_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
             "runwayml/stable-diffusion-v1-5").to(device)
 
-    # Optimize for ZeroGPU memory
+    # Optimize for ZeroGPU memory    
     sd_pipe.enable_attention_slicing()
+
+    # Do we need this conditionalize ?
+    sd_pipe = accelerator.prepare(sd_pipe)
     sd_pipe.enable_model_cpu_offload()
 
     return face_enhancer, sd_pipe

requirements.txt

@@ -41,4 +41,6 @@ yapf==0.40.2
 filterpy==1.4.5
 
 # Optional: For better performance
-# onnxruntime-gpu==1.17.0
+# onnxruntime-gpu==1.17.0
+
+accelerate