vikhyatk commited on
Commit
2037538
·
verified ·
1 Parent(s): 419d075

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -16,7 +16,7 @@ model = AutoModelForCausalLM.from_pretrained(
16
  trust_remote_code=True,
17
  torch_dtype=torch.float16,
18
  device_map={"": "cuda"},
19
- revision="56a3adeae60809e4269c544cde376feb20637ee0"
20
  )
21
 
22
  def visualize_faces_and_gaze(face_boxes, gaze_points=None, image=None, show_plot=True):
 
16
  trust_remote_code=True,
17
  torch_dtype=torch.float16,
18
  device_map={"": "cuda"},
19
+ revision="64ce9d45e8603058826ab339d3bb0f48f7289b1f"
20
  )
21
 
22
  def visualize_faces_and_gaze(face_boxes, gaze_points=None, image=None, show_plot=True):