Update app.py

app.py

@@ -13,11 +13,19 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+from diffusers import AutoencoderKL, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 from typing import Tuple
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import paramiko
 
+torch.backends.cuda.matmul.allow_tf32 = True
+torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = True
+torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = True
+torch.backends.cudnn.allow_tf32 = True
+torch.backends.cudnn.deterministic = False
+torch.backends.cudnn.benchmark = True
+torch.set_float32_matmul_precision("highest")
+
 FTP_HOST = "1ink.us"
 FTP_USER = "ford442"
 FTP_PASS = "GoogleBez12!"
@@ -141,7 +149,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-@spaces.GPU(duration=45)
+@spaces.GPU(duration=60)
 def generate(
     model_choice: str,
     prompt: str,
@@ -356,7 +364,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 minimum=10,
                 maximum=1000,
                 step=10,
-                value=255,
+                value=275,
             )
 
     gr.Examples(