eleazhong commited on
Commit
f7e5a2f
·
verified ·
1 Parent(s): 11d0d1a

Update qwenimage/optimization.py

Browse files
Files changed (1) hide show
  1. qwenimage/optimization.py +1 -1
qwenimage/optimization.py CHANGED
@@ -76,7 +76,7 @@ def drain_module_parameters(module: torch.nn.Module):
76
 
77
 
78
  # @ftimed
79
- # @spaces.GPU(duration=1500)
80
  def optimize_pipeline_(
81
  pipeline: Callable[P, Any],
82
  cache_compiled=True,
 
76
 
77
 
78
  # @ftimed
79
+ @spaces.GPU(duration=1500)
80
  def optimize_pipeline_(
81
  pipeline: Callable[P, Any],
82
  cache_compiled=True,