KingNish commited on
Commit
7467f27
·
verified ·
1 Parent(s): 1fc20a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(pipeline.transformer, weights=qint4, exclude="proj_out")
84
- freeze(pipeline.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])
 
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])