From 6f80adfce551966423546bc91f1245e4ecfcb717 Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Sat, 16 Mar 2024 18:18:56 +0100
Subject: [PATCH] Fix for deprecated op

---
 T5/t5v11.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/T5/t5v11.py b/T5/t5v11.py
index 23876d0..794c861 100644
--- a/T5/t5v11.py
+++ b/T5/t5v11.py
@@ -6,7 +6,6 @@ https://github.com/comfyanonymous/ComfyUI/blob/master/comfy/sd1_clip.py
 import os
 
 from transformers import T5Tokenizer, T5EncoderModel, T5Config, modeling_utils
-import comfy.ops
 import torch
 import traceback
 import zipfile
@@ -45,9 +44,8 @@ class T5v11Model(torch.nn.Module):
                 )
             config = T5Config.from_json_file(textmodel_json_config)
             self.num_layers = config.num_hidden_layers
-            with comfy.ops.use_comfy_ops(device, dtype):
-                with modeling_utils.no_init_weights():
-                    self.transformer = T5EncoderModel(config)
+            with modeling_utils.no_init_weights():
+                self.transformer = T5EncoderModel(config)
 
         if freeze:
             self.freeze()