Skip to content

Commit

Permalink
Merge pull request #76 from jug-dev/fix-inpainting
Browse files Browse the repository at this point in the history
Fix for inpainting when loading "ALL MODELS"
  • Loading branch information
jug-dev authored Mar 6, 2023
2 parents f0c6f16 + 7b20825 commit 962b897
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions worker/bridge_data/stable_diffusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,6 +118,10 @@ def get_all_models(self):

models = sorted(list(set(models)))

# Add inpainting model if inpainting is enabled
if self.allow_painting:
models.append("stable_diffusion_inpainting")

# Save our models to our environmental variable
os.environ["ALL_MODELS"] = ",".join(models)

Expand Down

0 comments on commit 962b897

Please sign in to comment.