Fix model_management import so it doesn't get executed twice.

This commit is contained in:
comfyanonymous
2023-04-15 18:55:17 -04:00
parent 81d1f00df3
commit 73c3e11e83
7 changed files with 14 additions and 14 deletions

View File

@@ -4,7 +4,7 @@ import copy
import sd1_clip
import sd2_clip
import model_management
from comfy import model_management
from .ldm.util import instantiate_from_config
from .ldm.models.autoencoder import AutoencoderKL
import yaml
@@ -388,7 +388,7 @@ class CLIP:
return cond
def encode(self, text):
tokens = self.tokenizer.tokenize_with_weights(text)
tokens = self.tokenize(text)
return self.encode_from_tokens(tokens)
class VAE: