Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ temperature = 0.8 # 1.0 = no change, < 1.0 = less random, > 1.0 = more random, i
|
|
34 |
top_k = 200 # retain only the top_k most likely tokens, clamp others to have 0 probability
|
35 |
device = 'cpu' # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
|
36 |
dtype = 'bfloat16' # 'float32' or 'bfloat16' or 'float16'
|
37 |
-
out_dir = '
|
38 |
|
39 |
|
40 |
torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
|
|
|
34 |
top_k = 200 # retain only the top_k most likely tokens, clamp others to have 0 probability
|
35 |
device = 'cpu' # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
|
36 |
dtype = 'bfloat16' # 'float32' or 'bfloat16' or 'float16'
|
37 |
+
out_dir = 'Eml' # ignored if init_from is not 'resume'
|
38 |
|
39 |
|
40 |
torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
|