Skip to content

Commit 5842ef4

Browse files
authored
change dtype in audioldm (#2776)
CVS-163050
1 parent 734a84b commit 5842ef4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

notebooks/sound-generation-audioldm2/sound-generation-audioldm2.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -797,7 +797,7 @@
797797
" def __init__(self, encoder_ir, config):\n",
798798
" self.encoder = core.compile_model(encoder_ir, device.value)\n",
799799
" self.config = config\n",
800-
" self.dtype = self.config.torch_dtype\n",
800+
" self.dtype = torch.float32\n",
801801
"\n",
802802
" def __call__(self, input_ids, **_):\n",
803803
" last_hidden_state = self.encoder(input_ids)[0]\n",

0 commit comments

Comments
 (0)