Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
f4b5851
1
Parent(s):
3f40ddb
Update load_model.py
Browse files- load_model.py +1 -1
load_model.py
CHANGED
|
@@ -18,7 +18,7 @@ from typing import Any, Dict, Optional, Union
|
|
| 18 |
|
| 19 |
import numpy as np
|
| 20 |
import torch
|
| 21 |
-
|
| 22 |
from fairseq.dataclass.configs import CheckpointConfig
|
| 23 |
from fairseq.dataclass.utils import (
|
| 24 |
convert_namespace_to_omegaconf,
|
|
|
|
| 18 |
|
| 19 |
import numpy as np
|
| 20 |
import torch
|
| 21 |
+
from fairseq.data import data_utils
|
| 22 |
from fairseq.dataclass.configs import CheckpointConfig
|
| 23 |
from fairseq.dataclass.utils import (
|
| 24 |
convert_namespace_to_omegaconf,
|