Update mypy dependencies

.mypy.ini

@@ -1,3 +1,33 @@
 [mypy]
 
 exclude = venv
+
+[mypy-alpaca_lora_4bit.*]
+ignore_missing_imports = True
+
+[mypy-flash_attn.*]
+ignore_missing_imports = True
+
+[mypy-huggingface_hub]
+ignore_missing_imports = True
+
+[mypy-transformers.*]
+ignore_missing_imports = True
+
+[mypy-peft]
+ignore_missing_imports = True
+
+[mypy-bitsandbytes]
+ignore_missing_imports = True
+
+[mypy-datasets]
+ignore_missing_imports = True
+
+[mypy-fire]
+ignore_missing_imports = True
+
+[mypy-setuptools]
+ignore_missing_imports = True
+
+[mypy-addict]
+ignore_missing_imports = True

.pre-commit-config.yaml

@@ -26,6 +26,5 @@ repos:
     - id: mypy
       additional_dependencies:
         [
-        'fire',
+            'types-PyYAML',
-        'types-PyYAML'
         ]