diff --git a/hubconf.py b/hubconf.py
index 1e79a89f426..e89754d87f3 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -19,6 +19,7 @@
 from torchvision.models.mnasnet import mnasnet0_5, mnasnet0_75, mnasnet1_0, mnasnet1_3
 from torchvision.models.mobilenetv2 import mobilenet_v2
 from torchvision.models.mobilenetv3 import mobilenet_v3_large, mobilenet_v3_small
+from torchvision.models.optical_flow import raft_large, raft_small
 from torchvision.models.regnet import (
     regnet_y_400mf,
     regnet_y_800mf,