AbstractPhil commited on
Commit
7a9d45a
·
verified ·
1 Parent(s): fae3920

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="beeper_final.safetensors")
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)