Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -80,8 +80,8 @@ pipe = PixArtAlphaPipeline.from_pretrained(
|
|
80 |
use_safetensors=True,
|
81 |
).to("cuda:0")
|
82 |
pipe.text_encoder.to_bettertransformer()
|
83 |
-
quantize(
|
84 |
-
freeze(
|
85 |
|
86 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|
87 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|
|
|
80 |
use_safetensors=True,
|
81 |
).to("cuda:0")
|
82 |
pipe.text_encoder.to_bettertransformer()
|
83 |
+
quantize(pipe.transformer, weights=qint4, exclude="proj_out")
|
84 |
+
freeze(pipe.transformer)
|
85 |
|
86 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|
87 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|