Spaces:

TimurHromek
/

HROM-V1

Running

TimurHromek commited on May 5

Commit

5129b3d

verified ·

1 Parent(s): 2e4f8d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ tokenizer_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.6/tokeniz
 tokenizer = Tokenizer.from_file(tokenizer_file)
 # 3. Load model checkpoint
-checkpoint_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.6//HROM-V1.6.pt")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def load_model():

 tokenizer = Tokenizer.from_file(tokenizer_file)
 # 3. Load model checkpoint
+checkpoint_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.6/HROM-V1.6.pt")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def load_model():