Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ import random
|
|
72 |
from transformers import AutoModel
|
73 |
import torch
|
74 |
|
75 |
-
|
76 |
# import torch_npu
|
77 |
# from torch_npu.contrib import transfer_to_npu
|
78 |
|
|
|
72 |
from transformers import AutoModel
|
73 |
import torch
|
74 |
|
75 |
+
### Uncomment if you want to use Ascend NPUs
|
76 |
# import torch_npu
|
77 |
# from torch_npu.contrib import transfer_to_npu
|
78 |
|