readme

lucidrains · Jun 29, 2023 · dc9e7c5 · dc9e7c5
1 parent 0599b5e
commit dc9e7c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -118,7 +118,7 @@ text_to_semantic = TextToSemantic(
 
 # load the trained text-to-semantic transformer
 
-text_to_semantic.load_state_dict(torch.load('/path/to/trained/model.pt'))
+text_to_semantic.load('/path/to/trained/model.pt')
 
 # pass it into the soundstorm