aifeifei798 commited on
Commit
82063dc
·
verified ·
1 Parent(s): ab48047

Upload 6 files

Browse files
Files changed (1) hide show
  1. feifeilib/feifeifluxapi.py +5 -5
feifeilib/feifeifluxapi.py CHANGED
@@ -7,10 +7,10 @@ from feifeilib.feifeisharpened import feifeisharpened
7
 
8
  client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
9
  client.headers["x-use-cache"] = "0"
10
- MAX_SEED = np.iinfo(np.int32).maxv
11
 
12
- def feifeifluxapi(randomize_seed=False,seed=random.randint(0, MAX_SEED),prompt,quality_select=False,sharpened_select=False, styles_Radio=["(None)"],FooocusExpansion_select=False,height=1152,width=896,guidance_scale=3.5,num_strength=0.35):
13
- prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
14
  image = client.text_to_image(
15
  prompt=prompt,
16
  height=height,
@@ -18,5 +18,5 @@ def feifeifluxapi(randomize_seed=False,seed=random.randint(0, MAX_SEED),prompt,q
18
  guidance_scale=guidance_scale
19
  )
20
  if sharpened_select:
21
- feifeisharpened(image,num_strength)
22
- return image
 
7
 
8
  client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
9
  client.headers["x-use-cache"] = "0"
10
+ MAX_SEED = np.iinfo(np.int32).max
11
 
12
+ def feifeifluxapi(prompt, randomize_seed=False, seed=random.randint(0, MAX_SEED), quality_select=False, sharpened_select=False, styles_Radio=["(None)"], FooocusExpansion_select=False, height=1152, width=896, guidance_scale=3.5, num_strength=0.35):
13
+ prompt, generator = feifeiprompt(randomize_seed, seed, prompt, quality_select, styles_Radio, FooocusExpansion_select)
14
  image = client.text_to_image(
15
  prompt=prompt,
16
  height=height,
 
18
  guidance_scale=guidance_scale
19
  )
20
  if sharpened_select:
21
+ feifeisharpened(image, num_strength)
22
+ return image