Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,10 @@ def video_to_audio_fn(video, prompt, neg_prompt, seed, num_steps, guidance, dura
|
|
| 102 |
REPO_ID = "LTTEAM/Nhep_Mieng"
|
| 103 |
snapshot_download(repo_id=REPO_ID, local_dir="checkpoints")
|
| 104 |
|
| 105 |
-
|
|
|
|
|
|
|
|
|
|
| 106 |
vae = AutoencoderKL.from_pretrained(
|
| 107 |
"stabilityai/sd-vae-ft-mse",
|
| 108 |
torch_dtype=torch.float16 if device=="cuda" else torch.float32
|
|
|
|
| 102 |
REPO_ID = "LTTEAM/Nhep_Mieng"
|
| 103 |
snapshot_download(repo_id=REPO_ID, local_dir="checkpoints")
|
| 104 |
|
| 105 |
+
BASE_DIR = os.path.dirname(__file__) # n岷縰 ch瓢a c贸
|
| 106 |
+
conf_path = os.path.join(BASE_DIR, "LatentSync", "configs", "unet", "second_stage.yaml")
|
| 107 |
+
conf = OmegaConf.load(conf_path)
|
| 108 |
+
|
| 109 |
vae = AutoencoderKL.from_pretrained(
|
| 110 |
"stabilityai/sd-vae-ft-mse",
|
| 111 |
torch_dtype=torch.float16 if device=="cuda" else torch.float32
|