Update fibo_vlm_prompt_to_json.py
Browse files
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/
|
| 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")
|