Update README.md
Browse files
README.md
CHANGED
@@ -169,7 +169,7 @@ print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=True))
|
|
169 |
### CUDA Info
|
170 |
|
171 |
- CUDA Version: 12.0
|
172 |
-
-
|
173 |
- Max Memory: {0: "37GB"}
|
174 |
- Device Map: {"": 0}
|
175 |
|
|
|
169 |
### CUDA Info
|
170 |
|
171 |
- CUDA Version: 12.0
|
172 |
+
- Hardware: 1 A100-SXM
|
173 |
- Max Memory: {0: "37GB"}
|
174 |
- Device Map: {"": 0}
|
175 |
|