JiantaoLin commited on
Commit
6db9d51
·
1 Parent(s): a2b7474
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -152,7 +152,7 @@ def text_to_detailed(prompt, seed=None):
152
  return k3d_wrapper.get_detailed_prompt(prompt, seed)
153
 
154
  @spaces.GPU(duration=120)
155
- def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_steps=20, redux_hparam=None, init_image=None, **kwargs):
156
  # subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
157
  # print(f"Before text_to_image: {torch.cuda.memory_allocated() / 1024**3} GB")
158
  # k3d_wrapper.flux_pipeline.enable_xformers_memory_efficient_attention()
 
152
  return k3d_wrapper.get_detailed_prompt(prompt, seed)
153
 
154
  @spaces.GPU(duration=120)
155
+ def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_steps=18, redux_hparam=None, init_image=None, **kwargs):
156
  # subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
157
  # print(f"Before text_to_image: {torch.cuda.memory_allocated() / 1024**3} GB")
158
  # k3d_wrapper.flux_pipeline.enable_xformers_memory_efficient_attention()