kfirbria commited on
Commit
45e2492
·
verified ·
1 Parent(s): e4eb593

Update fibo_vlm_prompt_to_json.py

Browse files
Files changed (1) hide show
  1. fibo_vlm_prompt_to_json.py +1 -1
fibo_vlm_prompt_to_json.py CHANGED
@@ -331,7 +331,7 @@ Editing instructions:
331
  class BriaFiboVLMPromptToJson(ModularPipelineBlocks):
332
  model_name = "BriaFibo"
333
 
334
- def __init__(self, model_id="briaai/vlm-processor-new"):
335
  super().__init__()
336
  self.engine = TransformersEngine(model_id)
337
  self.engine.model.to("cuda")
 
331
  class BriaFiboVLMPromptToJson(ModularPipelineBlocks):
332
  model_name = "BriaFibo"
333
 
334
+ def __init__(self, model_id="briaai/FIBO-vlm"):
335
  super().__init__()
336
  self.engine = TransformersEngine(model_id)
337
  self.engine.model.to("cuda")