Add List import to model.py for type hinting

This commit is contained in:
jayeshthk 2025-06-19 17:33:52 +05:30
parent b53e984052
commit 735546a4f9

View File

@ -2,7 +2,7 @@ import torch
import torch.nn as nn
import torch.nn.functional as F
import math
from typing import Optional, Tuple
from typing import Optional, Tuple, List
from dataclasses import dataclass
@dataclass