jedyang97 commited on
Commit
15bf65b
1 Parent(s): 252ad67

delete cuda line

Browse files
Files changed (1) hide show
  1. model.py +0 -1
model.py CHANGED
@@ -14,7 +14,6 @@ from llava.utils import disable_torch_init
14
 
15
  @spaces.GPU
16
  def load_model_and_dataloader(model_path, model_base, scene_to_obj_mapping, obj_context_feature_type="text", load_8bit=False, load_4bit=False, load_bf16=False):
17
- disable_torch_init()
18
 
19
  model_name = get_model_name_from_path(model_path)
20
 
 
14
 
15
  @spaces.GPU
16
  def load_model_and_dataloader(model_path, model_base, scene_to_obj_mapping, obj_context_feature_type="text", load_8bit=False, load_4bit=False, load_bf16=False):
 
17
 
18
  model_name = get_model_name_from_path(model_path)
19