Spaces:
Running
on
Zero
Running
on
Zero
shuanholmes
commited on
Commit
·
29a2fa1
1
Parent(s):
a11dece
[FireFlow] Fix Bug
Browse files
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
|