Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,231 Bytes
19020c2 65c1bb9 65655fc 19020c2 31ef714 19020c2 65655fc 19020c2 edb9151 31ef714 19020c2 31ef714 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import gradio as gr
import spaces
import random
import numpy as np
from extras.expansion import FooocusExpansion
from feifeilib.feifeimodload import feifeimodload
from feifeilib.feifeiprompt import feifeiprompt
from feifeilib.feifeisharpened import feifeisharpened
pipe = feifeimodload()
MAX_SEED = np.iinfo(np.int32).max
@spaces.GPU()
def feifeitexttoimg(
prompt,
quality_select=False,
sharpened_select=False,
styles_Radio=["(None)"],
FooocusExpansion_select=False,
seed=random.randint(0, MAX_SEED),
randomize_seed=False,
width=896,
height=1152,
num_inference_steps=4,
guidance_scale=3.5,
num_strength=0.35,
progress=gr.Progress(track_tqdm=True),
):
prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
image = pipe(
prompt="",
prompt_2=prompt,
width=width,
height=height,
num_inference_steps=num_inference_steps,
generator=generator,
guidance_scale=guidance_scale,
output_type="pil",
).images[0]
if sharpened_select:
feifeisharpened(image,num_strength)
return image
|