MohamedRashad commited on
Commit
e40c773
·
1 Parent(s): bdb3667

Update SDK version in README and adjust GPU decorator placement in app.py

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🏆
4
  colorFrom: purple
5
  colorTo: indigo
6
  sdk: gradio
7
- sdk_version: 5.25.2
8
  app_file: app.py
9
  pinned: false
10
  ---
 
4
  colorFrom: purple
5
  colorTo: indigo
6
  sdk: gradio
7
+ sdk_version: 5.23.0
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -64,6 +64,7 @@ stream = AsyncStream()
64
  outputs_folder = './outputs/'
65
  os.makedirs(outputs_folder, exist_ok=True)
66
 
 
67
  @torch.no_grad()
68
  def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
69
  total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
@@ -270,7 +271,6 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind
270
  stream.output_queue.push(('end', None))
271
  return
272
 
273
- @spaces.GPU(duration=120)
274
  def process(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
275
  global stream
276
  assert input_image is not None, 'No input image!'
 
64
  outputs_folder = './outputs/'
65
  os.makedirs(outputs_folder, exist_ok=True)
66
 
67
+ @spaces.GPU(duration=120)
68
  @torch.no_grad()
69
  def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
70
  total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
 
271
  stream.output_queue.push(('end', None))
272
  return
273
 
 
274
  def process(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
275
  global stream
276
  assert input_image is not None, 'No input image!'