This commit is contained in:
YaoyaoChang
2025-08-28 01:28:27 -07:00
parent 9537bd96a6
commit 4b8b6f7700
2 changed files with 68 additions and 68 deletions
+1 -1
View File
@@ -256,7 +256,7 @@ def main():
except Exception as e:
print(f"[ERROR] : {type(e).__name__}: {e}")
print(traceback.format_exc())
print("Error loading model, try sdpa.")
print("Error loading the model. Trying to use SDPA. However, note that only flash_attention_2 has been fully tested, and using SDPA may result in lower audio quality.")
model = VibeVoiceForConditionalGenerationInference.from_pretrained(
args.model_path,
torch_dtype=torch.bfloat16,