Update base/app.py

base/app.py

@@ -14,8 +14,8 @@ config_path = "./base/configs/sample.yaml"
 args = OmegaConf.load("./base/configs/sample.yaml")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 ### download models
-snapshot_download('Vchitect/LaVie',cache_dir='./pretrained_models')
-snapshot_download('CompVis/stable-diffusion-v1-4',cache_dir='./pretrained_models')
+# snapshot_download('Vchitect/LaVie',cache_dir='./pretrained_models')
+# snapshot_download('CompVis/stable-diffusion-v1-4',cache_dir='./pretrained_models')
 
 # ------- get model ---------------
 model_t2V = model_t2v_fun(args)