Spaces:

Vchitect
/

LaVie

Running on A10G

YaohuiW commited on Dec 1, 2023

Commit

6eb6fe1

•

1 Parent(s): cfeaf61

Update base/app.py

Files changed (1) hide show

base/app.py CHANGED Viewed

@@ -9,6 +9,7 @@ import pandas as pd
 import torchvision
 import random
 from huggingface_hub import snapshot_download
 config_path = "./base/configs/sample.yaml"
 args = OmegaConf.load("./base/configs/sample.yaml")

 import torchvision
 import random
 from huggingface_hub import snapshot_download
+from models import get_models
 config_path = "./base/configs/sample.yaml"
 args = OmegaConf.load("./base/configs/sample.yaml")