deepseekmirror/DeepSeek-Math
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Math.git synced 2025-06-21 00:43:45 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add List import to model.py for type hinting

Browse Source
This commit is contained in:
jayeshthk 2025-06-19 17:33:52 +05:30
parent b53e984052
commit 735546a4f9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train/model.py
View File

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