feat : cleaner import arrangement

2025-04-01 13:10:06 +02:00
parent b3efd09fb3
commit 1b5a55ccf2
8 changed files with 30 additions and 19 deletions
@@ -1,11 +1,12 @@
-import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import time
 import datetime
 import uuid
 import os
 import sys
 import json
+from typing import List, Tuple, Type, Dict, Tuple
+import torch
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))