Update README.md
Browse files
README.md
CHANGED
|
@@ -14,6 +14,8 @@ tags:
|
|
| 14 |
- 200m
|
| 15 |
---
|
| 16 |
|
|
|
|
|
|
|
| 17 |
# Dront 200m is a GPT-like model trained from scratch on approximately 5 billion tokens.
|
| 18 |
|
| 19 |
Context: 1024 tokens
|
|
@@ -105,7 +107,7 @@ libraries, and tools for creating and deploying programs.
|
|
| 105 |
```
|
| 106 |
|
| 107 |
|
| 108 |
-
# Easy Start
|
| 109 |
|
| 110 |
```Python
|
| 111 |
|
|
@@ -120,7 +122,7 @@ torch.backends.cudnn.benchmark = True
|
|
| 120 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 121 |
torch.set_float32_matmul_precision('high')
|
| 122 |
|
| 123 |
-
MODEL_NAME = "Dront-200m"
|
| 124 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 125 |
|
| 126 |
print("Loading model...")
|
|
@@ -183,4 +185,6 @@ while True:
|
|
| 183 |
print(f"Error: {e}")
|
| 184 |
continue
|
| 185 |
|
| 186 |
-
```
|
|
|
|
|
|
|
|
|
| 14 |
- 200m
|
| 15 |
---
|
| 16 |
|
| 17 |
+
|
| 18 |
+
|
| 19 |
# Dront 200m is a GPT-like model trained from scratch on approximately 5 billion tokens.
|
| 20 |
|
| 21 |
Context: 1024 tokens
|
|
|
|
| 107 |
```
|
| 108 |
|
| 109 |
|
| 110 |
+
# Easy Start ( ! Local !)
|
| 111 |
|
| 112 |
```Python
|
| 113 |
|
|
|
|
| 122 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 123 |
torch.set_float32_matmul_precision('high')
|
| 124 |
|
| 125 |
+
MODEL_NAME = "Dront-200m" # LOCAL FOLDER
|
| 126 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 127 |
|
| 128 |
print("Loading model...")
|
|
|
|
| 185 |
print(f"Error: {e}")
|
| 186 |
continue
|
| 187 |
|
| 188 |
+
```
|
| 189 |
+
|
| 190 |
+
Bye!
|