diff --git a/assets/caution.png b/assets/web/caution.png
similarity index 100%
rename from assets/caution.png
rename to assets/web/caution.png
diff --git a/ldm/invoke/ckpt_generator/base.py b/ldm/invoke/ckpt_generator/base.py
index 9d137b74d6..c066e19e5b 100644
--- a/ldm/invoke/ckpt_generator/base.py
+++ b/ldm/invoke/ckpt_generator/base.py
@@ -21,7 +21,7 @@ from ldm.models.diffusion.cross_attention_map_saving import AttentionMapSaver
 from ldm.util import rand_perlin_2d
 
 downsampling = 8
-CAUTION_IMG = 'assets/caution.png'
+CAUTION_IMG = 'assets/web/caution.png'
 
 class CkptGenerator():
     def __init__(self, model, precision):
diff --git a/ldm/invoke/generator/base.py b/ldm/invoke/generator/base.py
index a17badd022..9692f6c915 100644
--- a/ldm/invoke/generator/base.py
+++ b/ldm/invoke/generator/base.py
@@ -23,7 +23,7 @@ from ldm.models.diffusion.ddpm import DiffusionWrapper
 from ldm.util import rand_perlin_2d
 
 downsampling = 8
-CAUTION_IMG = 'assets/caution.png'
+CAUTION_IMG = 'assets/web/caution.png'
 
 class Generator:
     downsampling_factor: int
@@ -345,4 +345,3 @@ class Generator:
 
     def torch_dtype(self)->torch.dtype:
         return torch.float16 if self.precision == 'float16' else torch.float32
-
diff --git a/pyproject.toml b/pyproject.toml
index 0f2f34b962..b70cfb1685 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -108,10 +108,10 @@ version = { attr = "ldm.invoke.__version__" }
 
 [tool.setuptools.packages.find]
 "where" = ["."]
-"include" = ["assets", "backend*", "frontend.dist*", "ldm*"]
+"include" = ["assets.web*", "backend*", "frontend.dist*", "ldm*"]
 
 [tool.setuptools.package-data]
-"assets" = ["caution.png"]
+"assets.web" = ["**.png"]
 "backend" = ["**.png"]
 "frontend.dist" = ["**"]