Update open_flamingo/open_flamingo/src/factory.py

open_flamingo/open_flamingo/src/factory.py

@@ -59,9 +59,7 @@ def create_model_and_transforms(
     lang_encoder = AutoModelForCausalLM.from_pretrained(
         lang_encoder_path,
         local_files_only=use_local_files,
-        trust_remote_code=True,
-        
-    ).to(device, dtype=torch.bfloat16) if device > -1 else None
+        trust_remote_code=True).to(device, dtype=torch.bfloat16) if device > -1 else None
 
     # hacks for MPT-1B, which doesn't have a get_input_embeddings method
     if "mpt-1b-redpajama-200b" in lang_encoder_path:
@@ -92,8 +90,7 @@ def create_model_and_transforms(
             "width"
         ],
         cross_attn_every_n_layers=cross_attn_every_n_layers,
-        **flamingo_kwargs,
-    ).to(device, dtype=torch.bfloat16) if device > -1 else None
+        **flamingo_kwargs).to(device, dtype=torch.bfloat16) if device > -1 else None
 
     # Freeze all parameters
     model.requires_grad_(False)