Spaces:
Running
on
Zero
Running
on
Zero
Update demo.py
Browse files
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="
|
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):
|