Spaces:
Running on Zero

Ruurd commited on
Commit
d83b96d
·
verified ·
1 Parent(s): acc4845

Fix model repo name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,10 +36,10 @@ assistant_marker_ids = tokenizer.encode("Assistant:", add_special_tokens=False)
36
 
37
  def load_model():
38
  ckpt_path = hf_hub_download(
39
- repo_id="ruurd/tini",
40
  filename="diffusion-model.pth",
41
  token=os.getenv("HF_TOKEN"),
42
- # revision="5a22a8b6168466dbbf704efd00d8cbf2eee51426",
43
  )
44
 
45
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
36
 
37
  def load_model():
38
  ckpt_path = hf_hub_download(
39
+ repo_id="ruurd/tini_model",
40
  filename="diffusion-model.pth",
41
  token=os.getenv("HF_TOKEN"),
42
+ # revision="xxx",
43
  )
44
 
45
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")