diff --git a/gradio_app.py b/gradio_app.py index 47ec574..46b219d 100644 --- a/gradio_app.py +++ b/gradio_app.py @@ -748,8 +748,7 @@ if __name__ == '__main__': parser.add_argument('--compile', action='store_true') parser.add_argument('--low_vram_mode', action='store_true') args = parser.parse_args() - args.enable_flashvdm = False - + SAVE_DIR = args.cache_path os.makedirs(SAVE_DIR, exist_ok=True)