Update app.py
Browse files
app.py
CHANGED
@@ -789,11 +789,6 @@ def run_demo_server(pipe):
|
|
789 |
)
|
790 |
|
791 |
|
792 |
-
def prefetch_hf_cache(pipe):
|
793 |
-
process_image(pipe, "files/image/bee.jpg", 1, 1, 64)
|
794 |
-
shutil.rmtree("files/image/bee_output")
|
795 |
-
|
796 |
-
|
797 |
def main():
|
798 |
CHECKPOINT = "prs-eth/marigold-v1-0"
|
799 |
CHECKPOINT_UNET_LCM = "prs-eth/marigold-lcm-v1-0"
|
@@ -818,7 +813,6 @@ def main():
|
|
818 |
pass # run without xformers
|
819 |
|
820 |
pipe = pipe.to(device)
|
821 |
-
prefetch_hf_cache(pipe)
|
822 |
run_demo_server(pipe)
|
823 |
|
824 |
|
|
|
789 |
)
|
790 |
|
791 |
|
|
|
|
|
|
|
|
|
|
|
792 |
def main():
|
793 |
CHECKPOINT = "prs-eth/marigold-v1-0"
|
794 |
CHECKPOINT_UNET_LCM = "prs-eth/marigold-lcm-v1-0"
|
|
|
813 |
pass # run without xformers
|
814 |
|
815 |
pipe = pipe.to(device)
|
|
|
816 |
run_demo_server(pipe)
|
817 |
|
818 |
|