Update open_flamingo/open_flamingo/src/factory.py

open_flamingo/open_flamingo/src/factory.py

@@ -40,7 +40,7 @@ def create_model_and_transforms(
     )
     # set the vision encoder to output the visual features
     vision_encoder.visual.output_tokens = True
-        vision_encoder.to(device, dtype=torch.bfloat16) if device > -1 else None
+    vision_encoder.to(device, dtype=torch.bfloat16) if device > -1 else None
 
     text_tokenizer = AutoTokenizer.from_pretrained(
         tokenizer_path,