Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
191990f
1
Parent(s):
06fd2b2
Update t2i_client initialization to use the merged model and reduce inference steps
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ from trellis.utils import render_utils, postprocessing_utils
|
|
17 |
from gradio_client import Client
|
18 |
|
19 |
llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
|
20 |
-
t2i_client = Client("
|
21 |
|
22 |
def generate_t2i_prompt(item_name):
|
23 |
llm_prompt_template = """You are tasked with creating a concise yet highly detailed description of an item to be used for generating an image in a game development pipeline. The image should show the **entire item** with no parts cropped or hidden. The background should always be plain and monocolor, with no focus on it.
|
@@ -72,7 +72,7 @@ def generate_item_image(object_t2i_prompt):
|
|
72 |
width=1024,
|
73 |
height=1024,
|
74 |
guidance_scale=3.5,
|
75 |
-
num_inference_steps=
|
76 |
api_name="/infer"
|
77 |
)[0]
|
78 |
image = Image.open(img_path)
|
|
|
17 |
from gradio_client import Client
|
18 |
|
19 |
llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
|
20 |
+
t2i_client = Client("multimodalart/FLUX.1-merged")
|
21 |
|
22 |
def generate_t2i_prompt(item_name):
|
23 |
llm_prompt_template = """You are tasked with creating a concise yet highly detailed description of an item to be used for generating an image in a game development pipeline. The image should show the **entire item** with no parts cropped or hidden. The background should always be plain and monocolor, with no focus on it.
|
|
|
72 |
width=1024,
|
73 |
height=1024,
|
74 |
guidance_scale=3.5,
|
75 |
+
num_inference_steps=8,
|
76 |
api_name="/infer"
|
77 |
)[0]
|
78 |
image = Image.open(img_path)
|