diff --git a/torchchat/model.py b/torchchat/model.py
index 25b4ddcd7..7868b6593 100644
--- a/torchchat/model.py
+++ b/torchchat/model.py
@@ -12,8 +12,6 @@
 from enum import Enum
 from pathlib import Path
 
-import torchvision
-
 from typing import Any, Callable, Dict, Optional, Union
 from collections.abc import Hashable