Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ PREVIEW_IMAGES = True
|
|
29 |
dtype = torch.float16
|
30 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
31 |
if torch.cuda.is_available():
|
32 |
-
prior_pipeline = WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior-model-
|
33 |
decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=dtype)
|
34 |
if ENABLE_CPU_OFFLOAD:
|
35 |
prior_pipeline.enable_model_cpu_offload()
|
@@ -68,7 +68,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
68 |
|
69 |
def generate(
|
70 |
prompt: str,
|
71 |
-
negative_prompt: str = "",
|
72 |
seed: int = 0,
|
73 |
width: int = 1024,
|
74 |
height: int = 1024,
|
@@ -141,7 +141,7 @@ with gr.Blocks(css="style.css") as demo:
|
|
141 |
label="What I do NOT want",
|
142 |
max_lines=1,
|
143 |
placeholder="Uncheck seed to iterate and finetune.",
|
144 |
-
value="
|
145 |
)
|
146 |
|
147 |
seed = gr.Slider(
|
|
|
29 |
dtype = torch.float16
|
30 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
31 |
if torch.cuda.is_available():
|
32 |
+
prior_pipeline = WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior-model-base", torch_dtype=dtype)
|
33 |
decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=dtype)
|
34 |
if ENABLE_CPU_OFFLOAD:
|
35 |
prior_pipeline.enable_model_cpu_offload()
|
|
|
68 |
|
69 |
def generate(
|
70 |
prompt: str,
|
71 |
+
negative_prompt: str = "bad anatomy, blurry, fuzzy, extra arms, extra fingers, poorly drawn hands, disfigured, tiling, deformed, mutated, drawing, imperfections",
|
72 |
seed: int = 0,
|
73 |
width: int = 1024,
|
74 |
height: int = 1024,
|
|
|
141 |
label="What I do NOT want",
|
142 |
max_lines=1,
|
143 |
placeholder="Uncheck seed to iterate and finetune.",
|
144 |
+
value="Example: Text"
|
145 |
)
|
146 |
|
147 |
seed = gr.Slider(
|