Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
27 |
|
28 |
# Load weights from Hugging Face repo if not available locally
|
29 |
repo_id = "AbstractPhil/beeper-rose-tinystories-6l-512d-ctx512"
|
30 |
-
model_file = hf_hub_download(repo_id=repo_id, filename="
|
31 |
tokenizer_file = hf_hub_download(repo_id=repo_id, filename="tokenizer.json")
|
32 |
|
33 |
infer = BeeperRoseGPT(config).to(device)
|
|
|
27 |
|
28 |
# Load weights from Hugging Face repo if not available locally
|
29 |
repo_id = "AbstractPhil/beeper-rose-tinystories-6l-512d-ctx512"
|
30 |
+
model_file = hf_hub_download(repo_id=repo_id, filename="beeper_rose_final.safetensors")
|
31 |
tokenizer_file = hf_hub_download(repo_id=repo_id, filename="tokenizer.json")
|
32 |
|
33 |
infer = BeeperRoseGPT(config).to(device)
|