deepseekmirror/DeepSeek-V3
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-V3.git synced 2025-02-23 06:08:58 -05:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix small error

Browse Source
This commit is contained in:
Nripesh Niketan 2025-01-30 16:04:00 +00:00 committed by GitHub
parent e75ce46245
commit b6e3910fd0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference/model.py
View File

@ -802,7 +802,7 @@ if __name__ == "__main__":
default_device = "mps"
else:
default_device = "cpu"
torch.set_default_device("default_device")
torch.set_default_device(default_device)
torch.manual_seed(0)
args = ModelArgs()
x = torch.randint(0, args.vocab_size, (2, 128))