update

app.py

@@ -244,7 +244,7 @@ def create_training_demo(trainer: Trainer,
 def find_weight_files() -> list[str]:
     curr_dir = pathlib.Path(__file__).parent
     paths = sorted(curr_dir.rglob('*.bin'))
-    paths = [path for path in paths if '.lfs' not in path.name]
+    paths = [path for path in paths if '.lfs' not in str(path)]
     return [path.relative_to(curr_dir).as_posix() for path in paths]
 
 
@@ -298,14 +298,14 @@ def create_inference_demo(pipe: InferencePipeline) -> gr.Blocks:
                                                minimum=0,
                                                maximum=10.,
                                                step=1,
-                                               value=2)
+                                               value=1)
 
                 run_button = gr.Button('Generate')
 
                 gr.Markdown('''
                 - Models with names starting with "custom-diffusion-models/" are the pretrained models provided in the [original repo](https://github.com/adobe-research/custom-diffusion), and the ones with names starting with "results/delta.bin" are your trained models.
                 - After training, you can press "Reload Weight List" button to load your trained model names.
-                - Change default batch-size and steps for faster sampling. 
+                - Change batch-size and number of steps for more and better samples. 
                 ''')
             with gr.Column():
                 result = gr.Image(label='Result')

custom-diffusion-models/barn.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1e6ffb7953740286e005fb5ceffc3e985f93b3de97cd46202cf7d66d2171094b
-size 76690626
+oid sha256:b200774e6929f9f51d0c0f3f1e13455853a703e5552d563b7a7aa98ab6ff942c
+size 76690690

custom-diffusion-models/cat.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:08754e711b9ecaa36785dc64ad0c08317a93d106629c5f42cc5b9a406fe4aefc
-size 76690626
+oid sha256:93899d5900e9af2a7cff824146981194859d5c62c637c47ae6c487e16668aa6e
+size 76690690

custom-diffusion-models/chair.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3e2edb6eaddf540ab9e1a0aa75f3e46ee77c9ee41e8d8e87127777d5dd3ba4b7
-size 76690626
+oid sha256:a17c8e6b5b3d5f76c3b6792536113987a1de4958e107cfded31a6dd3f6d236b5
+size 76690690

custom-diffusion-models/dog.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dbe8a8843279fa01f2eaa3e9b0b34267e5b1949456f81b5bc17fb2a0d23086fe
-size 76690626
+oid sha256:bfacae58002ad0ac1a4e48bd98de9719a91038c4b55cd4caa721e3329460d6d4
+size 76690690

custom-diffusion-models/flower.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a37dcb14359c2baae984a218f62758eb52c842c7557e790063d4cd4daa120e5b
-size 76690626
+oid sha256:c10d2361abfe98728f647cdbd3fa0506e872ad65e9591d70a3cf2b0eb94f5cac
+size 76690690

custom-diffusion-models/table.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3771413ef6b319fd1f6df0ec2490febe17524cc3afd67b63042bed85af8cb9c2
-size 76690626
+oid sha256:ebea89e9d968f8bce8f06d9b881dacc5002edeca989a23ddf1473684219f87cb
+size 76690690

custom-diffusion-models/teddybear.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e0656c6eafed89a146f7ab971913c9ee35b4f0a96a4e1aa8eb8ccc28326a8164
-size 76690626
+oid sha256:c521048539a1b73a9d909f78f91fd776aa1441a12ae2358477660ad3eee1fcf0
+size 76690690

custom-diffusion-models/tortoise_plushy.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:38597922778e96c54e3a57379a11a356e62b76be990097922def3f9b764db48d
-size 76690626
+oid sha256:1f457414ca52d6ae71d96c432266c1ab4b53fe5c163055162a1f62318bc5f146
+size 76690690

custom-diffusion-models/wooden_pot.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:414cbd2bef3e7e65d860c4df17c8f8b8616f5dd8676634e0b228be8ed039f176
-size 76690626
+oid sha256:2affc64108eb7ba9c867fa3acbde15f8352b95bb24b0bb95dfeab913916d1db3
+size 76690690