Skip to content

Commit

Permalink
feat: only generate output when input image was changed
Browse files Browse the repository at this point in the history
  • Loading branch information
TimPietrusky committed Feb 22, 2024
1 parent 4feace7 commit f9f8cf0
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 2 deletions.
4 changes: 3 additions & 1 deletion .github/workflows/release-windows.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ on:

jobs:
release:
runs-on: windows-latest-l
runs-on: windows-latest
if: ${{ !contains(github.event.head_commit.message, 'chore(release)') }}
permissions:
contents: write # to be able to publish a GitHub release
Expand All @@ -18,6 +18,8 @@ jobs:
steps:
- name: Checkout
uses: actions/checkout@v3
with:
persist-credentials: false

- name: Setup node
uses: actions/setup-node@v4
Expand Down
13 changes: 12 additions & 1 deletion resources/python/live-painting/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,6 +239,9 @@ def main(pipe, input_image_path, output_image_path, shutdown_event):
strength = 0.99
guidance_scale = None

# When was the input image last modified
last_modified_time = None

# Queue to hold parameters received from stdin
params_queue = queue.Queue()

Expand All @@ -257,10 +260,18 @@ def main(pipe, input_image_path, output_image_path, shutdown_event):
seed = parameters.get("seed", seed)
strength = parameters.get("strength", strength)
guidance_scale = parameters.get("guidance_scale", guidance_scale)
print(f"Updated parameters {parameters}")
except queue.Empty:
pass # No new parameters, proceed with the existing ones

# Get the current modified time of the input image
current_modified_time = os.path.getmtime(input_image_path)

if current_modified_time != last_modified_time:
last_modified_time = current_modified_time
else:
# Skip this iteration since the input image has not changed
continue

# Only generate an image if the prompt is not empty
if prompt is not None and prompt.strip():
torch.manual_seed(seed)
Expand Down

0 comments on commit f9f8cf0

Please sign in to comment.