Spaces:
Running
on
Zero
Running
on
Zero
Update utils/t2i.py
Browse files- 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 |
-
|
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,
|