From 731525bb3e70c5a9bea66a7be7e8053cbe49055c Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 19 Jun 2024 21:00:45 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/nodes.py b/nodes.py
index e828a9f..aab295d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -31,10 +31,6 @@ class DownloadAndLoadFlorence2Model:
                     {
                     "default": 'microsoft/Florence-2-base'
                     }),
-            "precision": ([ 'bf16','fp32'],
-                    {
-                    "default": 'bf16'
-                    }),
             },
         }
 
@@ -43,10 +39,9 @@ class DownloadAndLoadFlorence2Model:
     FUNCTION = "loadmodel"
     CATEGORY = "Florence2"
 
-    def loadmodel(self, model, precision):
+    def loadmodel(self, model):
         device = mm.get_torch_device()
         offload_device = mm.unet_offload_device()
-        dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[precision]
 
         model_name = model.rsplit('/', 1)[-1]
         model_path = os.path.join(folder_paths.models_dir, "LLM", model_name)