More type fixes

monoxgas · monoxgas · commit a8ac4424b2b8 · 2025-05-08T00:13:31.000-07:00
diff --git a/rigging/generator/transformers_.py b/rigging/generator/transformers_.py
@@ -2,8 +2,8 @@
 import typing as t
 
 import torch
-import transformers
-from transformers import (
+import transformers  # type: ignore [import-untyped]
+from transformers import (  # type: ignore [import-untyped]
     AutoModelForCausalLM,
     AutoTokenizer,
     PreTrainedTokenizer,
