Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,25 +14,12 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
| 14 |
pipe.load_lora_weights("Keltezaa/NSFW_MASTER_FLUX")
|
| 15 |
pipe.to('cuda')
|
| 16 |
|
| 17 |
-
compel = Compel(
|
| 18 |
-
tokenizer=[pipe.tokenizer, pipe.tokenizer_2] ,
|
| 19 |
-
text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
|
| 20 |
-
returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
|
| 21 |
-
requires_pooled=[False, True]
|
| 22 |
-
)
|
| 23 |
-
|
| 24 |
|
| 25 |
@spaces.GPU(duration=120)
|
| 26 |
-
def generate(prompt,
|
| 27 |
-
embeds, pooled = compel(prompt)
|
| 28 |
-
neg_embeds, neg_pooled = compel(negative_prompt)
|
| 29 |
return pipe(
|
| 30 |
-
|
| 31 |
-
pooled_prompt_embeds=pooled,
|
| 32 |
-
negative_prompt_embeds=neg_embeds,
|
| 33 |
-
negative_pooled_prompt_embeds=neg_pooled,
|
| 34 |
num_inference_steps=num_inference_steps,
|
| 35 |
-
guidance_scale=guidance_scale,
|
| 36 |
width=width,
|
| 37 |
height=height,
|
| 38 |
num_images_per_prompt=num_samples
|
|
@@ -43,9 +30,7 @@ gr.Interface(
|
|
| 43 |
fn=generate,
|
| 44 |
inputs=[
|
| 45 |
gr.Text(label="Prompt"),
|
| 46 |
-
gr.Text("", label="Negative Prompt"),
|
| 47 |
gr.Number(7, label="Number inference steps"),
|
| 48 |
-
gr.Number(3, label="Guidance scale"),
|
| 49 |
gr.Number(512, label="Width"),
|
| 50 |
gr.Number(512, label="Height"),
|
| 51 |
gr.Number(1, label="# images"),
|
|
|
|
| 14 |
pipe.load_lora_weights("Keltezaa/NSFW_MASTER_FLUX")
|
| 15 |
pipe.to('cuda')
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
@spaces.GPU(duration=120)
|
| 19 |
+
def generate(prompt, num_inference_steps, width, height, num_samples):
|
|
|
|
|
|
|
| 20 |
return pipe(
|
| 21 |
+
prompt,
|
|
|
|
|
|
|
|
|
|
| 22 |
num_inference_steps=num_inference_steps,
|
|
|
|
| 23 |
width=width,
|
| 24 |
height=height,
|
| 25 |
num_images_per_prompt=num_samples
|
|
|
|
| 30 |
fn=generate,
|
| 31 |
inputs=[
|
| 32 |
gr.Text(label="Prompt"),
|
|
|
|
| 33 |
gr.Number(7, label="Number inference steps"),
|
|
|
|
| 34 |
gr.Number(512, label="Width"),
|
| 35 |
gr.Number(512, label="Height"),
|
| 36 |
gr.Number(1, label="# images"),
|