Spaces:
Sleeping
Sleeping
Update gpt_config.py
Browse files- gpt_config.py +2 -2
gpt_config.py
CHANGED
@@ -6,8 +6,8 @@ block_size = 256 # what is the maximum context length for predictions?
|
|
6 |
max_iters = 10000
|
7 |
eval_interval = 500
|
8 |
learning_rate = 3e-4
|
9 |
-
|
10 |
-
device = 'mps'
|
11 |
eval_iters = 200
|
12 |
n_embd = 384
|
13 |
n_head = 6
|
|
|
6 |
max_iters = 10000
|
7 |
eval_interval = 500
|
8 |
learning_rate = 3e-4
|
9 |
+
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
10 |
+
# device = 'mps'
|
11 |
eval_iters = 200
|
12 |
n_embd = 384
|
13 |
n_head = 6
|