mirror of
https://git.datalinker.icu/deepseek-ai/DeepSeek-V3.git
synced 2026-06-02 10:54:31 +08:00
Update generate.py
This commit is contained in:
parent
592fd5daf8
commit
2988c05daa
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user