erwold commited on
Commit
660497c
·
1 Parent(s): cec4374

Initial Commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -202,7 +202,7 @@ class FluxInterface:
202
 
203
  return pooled_prompt_embeds
204
 
205
- @spaces.GPU(duration=300) # 300秒的 GPU 使用时间
206
  def generate(self, input_image, prompt="", guidance_scale=3.5, num_inference_steps=28, num_images=2, seed=None, aspect_ratio="1:1"):
207
  try:
208
  logger.info(f"Starting generation with prompt: {prompt}, guidance_scale: {guidance_scale}, steps: {num_inference_steps}")
 
202
 
203
  return pooled_prompt_embeds
204
 
205
+ @spaces.GPU(duration=120) # 300秒的 GPU 使用时间
206
  def generate(self, input_image, prompt="", guidance_scale=3.5, num_inference_steps=28, num_images=2, seed=None, aspect_ratio="1:1"):
207
  try:
208
  logger.info(f"Starting generation with prompt: {prompt}, guidance_scale: {guidance_scale}, steps: {num_inference_steps}")