Spaces:

Syzygianinfern0
/

NeuS-V

Sleeping

Syzygianinfern0 commited on Apr 4

Commit

31d38b7

1 Parent(s): e526530

Fix device idx

Files changed (1) hide show

evaluate_demo.py CHANGED Viewed

@@ -22,7 +22,7 @@ WEIGHT_PATH = Path("./assets/")
 pickle_path = WEIGHT_PATH / "distributions.pkl"
 num_of_frame_in_sequence = 3
 model = "InternVL2-8B"
-device = 7
 # Load the vision-language model
 vision_language_model = InternVL(model_name=model, device=device)
 # Load distributions

 pickle_path = WEIGHT_PATH / "distributions.pkl"
 num_of_frame_in_sequence = 3
 model = "InternVL2-8B"
+device = 0
 # Load the vision-language model
 vision_language_model = InternVL(model_name=model, device=device)
 # Load distributions