vikhyatk commited on
Commit
688a87d
·
verified ·
1 Parent(s): e1beb1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  try:
2
  import spaces
3
 
@@ -50,15 +52,14 @@ if IN_SPACES:
50
  )
51
 
52
  os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
53
- revision = "b1a66922578c2f1b9c3b9370fe1dfa2b1c1d176b"
54
- tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream-next", revision=revision)
55
  moondream = AutoModelForCausalLM.from_pretrained(
56
  "vikhyatk/moondream-next",
57
  trust_remote_code=True,
58
  torch_dtype=torch.float16,
59
  device_map={"": "cuda"},
60
  attn_implementation="flash_attention_2",
61
- revision=revision
62
  )
63
 
64
  # CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]
 
1
+ REVISION = "d1014146e28f5c79a36d15b4173bebfe5bfcf333"
2
+
3
  try:
4
  import spaces
5
 
 
52
  )
53
 
54
  os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
55
+ tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream-next", revision=REVISION)
 
56
  moondream = AutoModelForCausalLM.from_pretrained(
57
  "vikhyatk/moondream-next",
58
  trust_remote_code=True,
59
  torch_dtype=torch.float16,
60
  device_map={"": "cuda"},
61
  attn_implementation="flash_attention_2",
62
+ revision=REVISION
63
  )
64
 
65
  # CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]