Update README.md
Browse fileschange the path slash
README.md
CHANGED
|
@@ -15,7 +15,7 @@ import torch
|
|
| 15 |
from transformers import AutoTokenizer, AutoModel
|
| 16 |
|
| 17 |
|
| 18 |
-
model_path = "cntd
|
| 19 |
print("是否可用:", torch.cuda.is_available()) # 查看GPU是否可用
|
| 20 |
print("GPU数量:", torch.cuda.device_count()) # 查看GPU数量
|
| 21 |
print("torch方法查看CUDA版本:", torch.version.cuda) # torch方法查看CUDA版本
|
|
|
|
| 15 |
from transformers import AutoTokenizer, AutoModel
|
| 16 |
|
| 17 |
|
| 18 |
+
model_path = "cntd/CNTDAI-6B"
|
| 19 |
print("是否可用:", torch.cuda.is_available()) # 查看GPU是否可用
|
| 20 |
print("GPU数量:", torch.cuda.device_count()) # 查看GPU数量
|
| 21 |
print("torch方法查看CUDA版本:", torch.version.cuda) # torch方法查看CUDA版本
|