Update generate.py

This commit is contained in:
Antonio Gallardo García 2025-03-13 13:38:21 +01:00 committed by GitHub
parent 592fd5daf8
commit 2988c05daa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,7 +1,7 @@
import os
import json
from argparse import ArgumentParser
from typing import List
from typing import List
import torch
import torch.distributed as dist