Add variant argument to support loading fp16 model

app.py

@@ -283,6 +283,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 pipe = ObjectClearPipeline.from_pretrained_with_custom_modules(
     "jixin0101/ObjectClear",
     torch_dtype=torch.float16,
+    variant='fp16',
     save_cross_attn=True,
     cache_dir="/home/jovyan/shared/jixinzhao/models",
 )

pipeline_objectclear.py

@@ -469,6 +469,7 @@ class ObjectClearPipeline(
         pretrained_model_name_or_path,
         torch_dtype=torch.float32,
         cache_dir=None,
+        variant=None,
         **kwargs,
     ):
         from safetensors.torch import load_file
@@ -497,6 +498,7 @@ class ObjectClearPipeline(
             image_prompt_encoder=image_prompt_encoder,
             postfuse_module=postfuse_module,
             cache_dir=cache_dir,
+            variant=variant,
             **kwargs,
         )