Spaces:
Runtime error
Runtime error
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -11,9 +11,10 @@ import gradio as gr
|
|
| 11 |
from nanochat.checkpoint_manager import load_model_from_dir
|
| 12 |
from nanochat.engine import Engine
|
| 13 |
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
|
|
|
| 17 |
|
| 18 |
ckpt_dir = f"/tmp/ckpt/d{DEPTH}"
|
| 19 |
os.makedirs(ckpt_dir, exist_ok=True)
|
|
|
|
| 11 |
from nanochat.checkpoint_manager import load_model_from_dir
|
| 12 |
from nanochat.engine import Engine
|
| 13 |
|
| 14 |
+
# Hardcoded model selection for this Space
|
| 15 |
+
MODEL_REPO = "loocorez/nanochat-base-d20-step21400"
|
| 16 |
+
STEP = "021400"
|
| 17 |
+
DEPTH = "20"
|
| 18 |
|
| 19 |
ckpt_dir = f"/tmp/ckpt/d{DEPTH}"
|
| 20 |
os.makedirs(ckpt_dir, exist_ok=True)
|