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 os
import json import json
from argparse import ArgumentParser from argparse import ArgumentParser
from typing import List from typing import List
import torch import torch
import torch.distributed as dist import torch.distributed as dist