Nupur Kumari commited on
Commit
294c8dc
1 Parent(s): ddb7b56
app.py CHANGED
@@ -244,7 +244,7 @@ def create_training_demo(trainer: Trainer,
244
  def find_weight_files() -> list[str]:
245
  curr_dir = pathlib.Path(__file__).parent
246
  paths = sorted(curr_dir.rglob('*.bin'))
247
- paths = [path for path in paths if '.lfs' not in path.name]
248
  return [path.relative_to(curr_dir).as_posix() for path in paths]
249
 
250
 
@@ -298,14 +298,14 @@ def create_inference_demo(pipe: InferencePipeline) -> gr.Blocks:
298
  minimum=0,
299
  maximum=10.,
300
  step=1,
301
- value=2)
302
 
303
  run_button = gr.Button('Generate')
304
 
305
  gr.Markdown('''
306
  - 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.
307
  - After training, you can press "Reload Weight List" button to load your trained model names.
308
- - Change default batch-size and steps for faster sampling.
309
  ''')
310
  with gr.Column():
311
  result = gr.Image(label='Result')
 
244
  def find_weight_files() -> list[str]:
245
  curr_dir = pathlib.Path(__file__).parent
246
  paths = sorted(curr_dir.rglob('*.bin'))
247
+ paths = [path for path in paths if '.lfs' not in str(path)]
248
  return [path.relative_to(curr_dir).as_posix() for path in paths]
249
 
250
 
 
298
  minimum=0,
299
  maximum=10.,
300
  step=1,
301
+ value=1)
302
 
303
  run_button = gr.Button('Generate')
304
 
305
  gr.Markdown('''
306
  - 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.
307
  - After training, you can press "Reload Weight List" button to load your trained model names.
308
+ - Change batch-size and number of steps for more and better samples.
309
  ''')
310
  with gr.Column():
311
  result = gr.Image(label='Result')
custom-diffusion-models/barn.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e6ffb7953740286e005fb5ceffc3e985f93b3de97cd46202cf7d66d2171094b
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b200774e6929f9f51d0c0f3f1e13455853a703e5552d563b7a7aa98ab6ff942c
3
+ size 76690690
custom-diffusion-models/cat.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:08754e711b9ecaa36785dc64ad0c08317a93d106629c5f42cc5b9a406fe4aefc
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93899d5900e9af2a7cff824146981194859d5c62c637c47ae6c487e16668aa6e
3
+ size 76690690
custom-diffusion-models/chair.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e2edb6eaddf540ab9e1a0aa75f3e46ee77c9ee41e8d8e87127777d5dd3ba4b7
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a17c8e6b5b3d5f76c3b6792536113987a1de4958e107cfded31a6dd3f6d236b5
3
+ size 76690690
custom-diffusion-models/dog.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dbe8a8843279fa01f2eaa3e9b0b34267e5b1949456f81b5bc17fb2a0d23086fe
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfacae58002ad0ac1a4e48bd98de9719a91038c4b55cd4caa721e3329460d6d4
3
+ size 76690690
custom-diffusion-models/flower.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a37dcb14359c2baae984a218f62758eb52c842c7557e790063d4cd4daa120e5b
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c10d2361abfe98728f647cdbd3fa0506e872ad65e9591d70a3cf2b0eb94f5cac
3
+ size 76690690
custom-diffusion-models/table.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3771413ef6b319fd1f6df0ec2490febe17524cc3afd67b63042bed85af8cb9c2
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebea89e9d968f8bce8f06d9b881dacc5002edeca989a23ddf1473684219f87cb
3
+ size 76690690
custom-diffusion-models/teddybear.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e0656c6eafed89a146f7ab971913c9ee35b4f0a96a4e1aa8eb8ccc28326a8164
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c521048539a1b73a9d909f78f91fd776aa1441a12ae2358477660ad3eee1fcf0
3
+ size 76690690
custom-diffusion-models/tortoise_plushy.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:38597922778e96c54e3a57379a11a356e62b76be990097922def3f9b764db48d
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f457414ca52d6ae71d96c432266c1ab4b53fe5c163055162a1f62318bc5f146
3
+ size 76690690
custom-diffusion-models/wooden_pot.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:414cbd2bef3e7e65d860c4df17c8f8b8616f5dd8676634e0b228be8ed039f176
3
- size 76690626
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2affc64108eb7ba9c867fa3acbde15f8352b95bb24b0bb95dfeab913916d1db3
3
+ size 76690690