Commit 1a435240 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix model loading twice in some situations

parent 5f5435eb
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -216,6 +216,9 @@ class StableDiffusionModelHijack:
        self.comments = []

    def get_prompt_lengths(self, text):
        if self.clip is None:
            return "-", "-"

        _, token_count = self.clip.process_texts([text])

        return token_count, self.clip.get_target_prompt_token_count(token_count)
+3 −0
Original line number Diff line number Diff line
@@ -414,6 +414,9 @@ class SdModelData:
    def get_sd_model(self):
        if self.sd_model is None:
            with self.lock:
                if self.sd_model is not None:
                    return self.sd_model

                try:
                    load_model()
                except Exception as e: