Hugging Face's logo

Spaces:
moondream
/
test-space
Running on Zero

App Files Community
vikhyatk commited on
Commit
718a94f
·
verified ·
1 Parent(s): 5091728

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -56,6 +56,7 @@ moondream = AutoModelForCausalLM.from_pretrained(
56
  torch_dtype=torch.float16,
57
  device_map={"": "cuda"},
58
  attn_implementation="flash_attention_2",
 
59
  )
60
 
61
  # CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]
 
56
  torch_dtype=torch.float16,
57
  device_map={"": "cuda"},
58
  attn_implementation="flash_attention_2",
59
+ revision="a23030ab157f2d0f5bb2df6c6d43623904727ec2",
60
  )
61
 
62
  # CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]