Fix for deprecated op

2024-03-16 18:18:56 +01:00
parent 8befcdd9d6
commit 6f80adfce5
1 changed files with 2 additions and 4 deletions
@@ -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()