Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import spaces
 import numpy as np
 import uuid
-from optimum.quanto import freeze, qfloat8, quantize, qint4
+from optimum.quanto import freeze, qfloat8, quantize
 from diffusers import PixArtAlphaPipeline, LCMScheduler
 import torch
 from typing import Tuple
@@ -79,8 +79,7 @@ pipe = PixArtAlphaPipeline.from_pretrained(
         torch_dtype=torch.float16,
         use_safetensors=True,
 ).to("cuda:0")
-pipe.text_encoder.to_bettertransformer()
-quantize(pipe.transformer, weights=qint4, exclude="proj_out")
+quantize(pipe.transformer, weights=qfloat8)
 freeze(pipe.transformer)
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]: