MohamedRashad commited on
Commit
191990f
·
1 Parent(s): 06fd2b2

Update t2i_client initialization to use the merged model and reduce inference steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("black-forest-labs/FLUX.1-schnell")
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=28,
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)