BricksDisplay
/

ellie-Bert-VITS2

feature-extraction

Model card Files Files and versions Community

hans00 commited on Jun 18

Commit

c4a5682

•

1 Parent(s): 510ec41

Correct example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -19,8 +19,8 @@ import torch
 model_name = "BricksDisplay/ellie-Bert-VITS2"
-model = AutoModel.from_pretrained(model_name, trust_remote=True)
-processor = AutoProcessor.from_pretrained(model_name, trust_remote=True)
 with torch.no_grad():
     inputs = processor("你好", language="zh", return_tensors="pt")

 model_name = "BricksDisplay/ellie-Bert-VITS2"
+model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
+processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
 with torch.no_grad():
     inputs = processor("你好", language="zh", return_tensors="pt")