shuanholmes commited on
Commit
29a2fa1
·
1 Parent(s): a11dece

[FireFlow] Fix Bug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ add_sampling_metadata = True
60
  @spaces.GPU(duration=120)
61
  @torch.inference_mode()
62
  def edit(init_image, source_prompt, target_prompt, editing_strategy, num_steps, inject_step, guidance, seed):
63
- global ae, t5, clip, model, name, is_schnell, output_dir, add_sampling_metadata
64
  device = "cuda" if torch.cuda.is_available() else "cpu"
65
  torch.cuda.empty_cache()
66
  seed = None
 
60
  @spaces.GPU(duration=120)
61
  @torch.inference_mode()
62
  def edit(init_image, source_prompt, target_prompt, editing_strategy, num_steps, inject_step, guidance, seed):
63
+ global ae, t5, clip, model, name, is_schnell, output_dir, add_sampling_metadata, offload
64
  device = "cuda" if torch.cuda.is_available() else "cpu"
65
  torch.cuda.empty_cache()
66
  seed = None