feat: run locally and change parameters

app.py

@@ -26,7 +26,6 @@ body {
     background-repeat: no-repeat;
     background-attachment: fixed;
     background-position: center;
-    backdrop-filter: blur(3px);
     min-height: 100vh;
 }
 #col-container {
@@ -74,7 +73,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML("""
                 <div align="center" style="margin-bottom: 20px;">
-                    <img src='Faice.png' width="160">
+                    <img src='/gradio_api/file=Faice.png' width="160">
                     <p style="font-size: 16px; max-width: 960px; margin: 5px auto;">
                         Human Faces Generation with Diffusion Models.
                     </p>

stable_diffusion_inference.py

@@ -18,7 +18,8 @@ from utils import timer
 
 model_path = 'Ngene787/Faice_text2face'
 
-accelerator = Accelerator(mixed_precision="fp16", gradient_accumulation_steps=1)
+# accelerator = Accelerator(mixed_precision="fp16", gradient_accumulation_steps=1)
+accelerator = Accelerator(gradient_accumulation_steps=1)
 logger.info("Loading model ...")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 if torch.cuda.is_available():
@@ -35,7 +36,7 @@ pipe = pipe.to(device)
 
 pipe = accelerator.prepare(pipe)
 # Enable memory-efficient attention
-pipe.enable_xformers_memory_efficient_attention()
+# pipe.enable_xformers_memory_efficient_attention()
 
 # Enable attention slicing
 pipe.enable_attention_slicing()