-
Notifications
You must be signed in to change notification settings - Fork 2
Description
Hi author, when I run your inference script, an error happened. I noticed that the current code is not completely uploaded. Can you update your current code?
Traceback (most recent call last):
File "/home/vsw/Desktop/BADAS-Open/main.py", line 3, in
model = BADASModel(device="cuda")
File "/home/vsw/Desktop/BADAS-Open/badas/badas_loader.py", line 71, in init
self.model = load_badas_model(device=self.device, checkpoint_path=checkpoint_path)
File "/home/vsw/Desktop/BADAS-Open/badas/badas_loader.py", line 50, in load_badas_model
model.load()
File "/home/vsw/Desktop/BADAS-Open/badas/models/vjepa.py", line 88, in load
raise RuntimeError(f"Failed to load V-JEPA model: {e}")
RuntimeError: Failed to load V-JEPA model: Training modules not available - cannot load V-JEPA model