eienmojiki commited on
Commit
2fbd841
·
verified ·
1 Parent(s): adee991

Update utils/t2i.py

Browse files
Files changed (1) hide show
  1. utils/t2i.py +2 -1
utils/t2i.py CHANGED
@@ -36,7 +36,7 @@ trans_vae.to(device)
36
  @spaces.GPU(duration=120)
37
  def t2i_gen(
38
  prompt: str,
39
- # negative_prompt: str = None,
40
  seed: int = 1111,
41
  width: int = 1024,
42
  height: int = 1024,
@@ -46,6 +46,7 @@ def t2i_gen(
46
  t2i_pipe.load_lora_weights("RedAIGC/Flux-version-LayerDiffuse", weight_name="layerlora.safetensors")
47
  latents = t2i_pipe(
48
  prompt=prompt,
 
49
  height=height,
50
  width=width,
51
  num_inference_steps=num_inference_steps,
 
36
  @spaces.GPU(duration=120)
37
  def t2i_gen(
38
  prompt: str,
39
+ negative_prompt: str = None,
40
  seed: int = 1111,
41
  width: int = 1024,
42
  height: int = 1024,
 
46
  t2i_pipe.load_lora_weights("RedAIGC/Flux-version-LayerDiffuse", weight_name="layerlora.safetensors")
47
  latents = t2i_pipe(
48
  prompt=prompt,
49
+ negative_prompt=negative_prompt,
50
  height=height,
51
  width=width,
52
  num_inference_steps=num_inference_steps,