Skip to content

GitLab

Unverified Commit 34308cc5 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix import error

parent d60766cb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
unicore/utils.py
View file @ 34308cc5
...@@ -13,6 +13,7 @@ import warnings ...@@ -13,6 +13,7 @@ import warnings
from functools import partial from functools import partial
from typing import List, Callable, Any, Dict from typing import List, Callable, Any, Dict
import torch import torch
import torch.utils.checkpoint
import torch.nn.functional as F import torch.nn.functional as F
try: try:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment