Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
|
|
6 |
import spaces
|
7 |
import numpy as np
|
8 |
import uuid
|
9 |
-
from diffusers import PixArtAlphaPipeline, LCMScheduler
|
10 |
import torch
|
11 |
from typing import Tuple
|
12 |
from datetime import datetime
|
@@ -80,8 +80,6 @@ pipe = PixArtAlphaPipeline.from_pretrained(
|
|
80 |
torch_dtype=torch.float16,
|
81 |
use_safetensors=True,
|
82 |
).to("cuda:0")
|
83 |
-
|
84 |
-
pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
|
85 |
pipe.text_encoder.to_bettertransformer()
|
86 |
|
87 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|
|
|
6 |
import spaces
|
7 |
import numpy as np
|
8 |
import uuid
|
9 |
+
from diffusers import PixArtAlphaPipeline, LCMScheduler
|
10 |
import torch
|
11 |
from typing import Tuple
|
12 |
from datetime import datetime
|
|
|
80 |
torch_dtype=torch.float16,
|
81 |
use_safetensors=True,
|
82 |
).to("cuda:0")
|
|
|
|
|
83 |
pipe.text_encoder.to_bettertransformer()
|
84 |
|
85 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|