Skip to content

Commit dc545b2

Browse files
authored
Merge pull request #242 from guynich/fix_model_definition
Fix undefined model
2 parents 17e2964 + c9dcea2 commit dc545b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Gemma/[Gemma_3n]Audio_understanding_with_HF.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@
147147
" )\n",
148148
" input_len = input_ids[\"input_ids\"].shape[-1]\n",
149149
"\n",
150-
" input_ids = input_ids.to(self.model.device, dtype=model.dtype)\n",
150+
" input_ids = input_ids.to(self.model.device, dtype=self.model.dtype)\n",
151151
" outputs = self.model.generate(\n",
152152
" **input_ids,\n",
153153
" max_new_tokens=max_tokens,\n",

0 commit comments

Comments
 (0)