Fix imports for trust_remote_code: mtp_loss.py
Browse files- mtp_loss.py +1 -1
mtp_loss.py
CHANGED
|
@@ -13,7 +13,7 @@ import torch
|
|
| 13 |
import torch.nn as nn
|
| 14 |
from typing import Tuple, Dict
|
| 15 |
|
| 16 |
-
from configuration_aether_micro import AETHERMicroConfig
|
| 17 |
|
| 18 |
|
| 19 |
class MTPLoss(nn.Module):
|
|
|
|
| 13 |
import torch.nn as nn
|
| 14 |
from typing import Tuple, Dict
|
| 15 |
|
| 16 |
+
from .configuration_aether_micro import AETHERMicroConfig
|
| 17 |
|
| 18 |
|
| 19 |
class MTPLoss(nn.Module):
|