diffsynth-engine 0.3.4__tar.gz → 0.3.6.dev2__tar.gz

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.3.4
+Version: 0.3.6.dev2
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/models/sd/sd_controlnet.py

@@ -10,8 +10,6 @@ from diffsynth_engine.models.basic.unet_helper import (
     AttentionBlock,
     PushBlock,
     DownSampler,
-    PopBlock,
-    UpSampler,
 )
 
 class ControlNetConditioningLayer(nn.Module):
@@ -565,7 +563,6 @@ class SDControlNet(PreTrainedModel):
         time_emb = self.time_embedding(timestep, dtype=sample.dtype)
 
         # 2. pre-process
-        height, width = sample.shape[2], sample.shape[3]
         hidden_states = self.conv_in(sample) + self.controlnet_conv_in(conditioning)
         text_emb = encoder_hidden_states
         res_stack = [hidden_states]

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/models/sdxl/sdxl_controlnet.py

@@ -1,13 +1,10 @@
 import torch
-import torch.nn as nn
 from typing import Optional, Dict
 from diffsynth_engine.models.basic.unet_helper import (
     ResnetBlock,
     AttentionBlock,
     PushBlock,
     DownSampler,
-    PopBlock,
-    UpSampler,
 )
 from diffsynth_engine.models.sd.sd_controlnet import ControlNetConditioningLayer
 from diffsynth_engine.models.base import PreTrainedModel, StateDictConverter
@@ -283,7 +280,6 @@ class SDXLControlNetUnion(PreTrainedModel):
         time_emb = t_emb + add_embeds + control_embeds
 
         # 2. pre-process
-        height, width = sample.shape[2], sample.shape[3]
         hidden_states = self.conv_in(sample)
         hidden_states = self.fuse_condition_to_input(hidden_states, task_id, conditioning)
         text_emb = encoder_hidden_states

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/pipelines/controlnet_helper.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from typing import Dict, List, Tuple, Union, Optional
+from typing import List, Union, Optional
 from PIL import Image
 from dataclasses import dataclass
 

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/pipelines/flux_image.py

@@ -2,12 +2,11 @@ import re
 import os
 import json
 import torch
-import torch.nn as nn
 import torch.distributed as dist
 import math
 from einops import rearrange
 from enum import Enum
-from typing import Callable, Dict, List, Tuple, Optional, Union
+from typing import Callable, Dict, List, Tuple, Optional
 from tqdm import tqdm
 from PIL import Image
 from dataclasses import dataclass

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/pipelines/sd_image.py

@@ -18,6 +18,7 @@ from diffsynth_engine.algorithm.noise_scheduler import ScaledLinearScheduler
 from diffsynth_engine.algorithm.sampler import EulerSampler
 from diffsynth_engine.utils.prompt import tokenize_long_prompt
 from diffsynth_engine.utils.constants import SDXL_TOKENIZER_CONF_PATH
+from diffsynth_engine.utils.platform import empty_cache
 from diffsynth_engine.utils import logging
 
 logger = logging.get_logger(__name__)

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine/pipelines/sdxl_image.py

@@ -26,6 +26,7 @@ from diffsynth_engine.algorithm.noise_scheduler import ScaledLinearScheduler
 from diffsynth_engine.algorithm.sampler import EulerSampler
 from diffsynth_engine.utils.prompt import tokenize_long_prompt
 from diffsynth_engine.utils.constants import SDXL_TOKENIZER_CONF_PATH, SDXL_TOKENIZER_2_CONF_PATH
+from diffsynth_engine.utils.platform import empty_cache
 from diffsynth_engine.utils import logging
 
 logger = logging.get_logger(__name__)
@@ -89,6 +90,8 @@ class SDXLLoRAConverter(LoRAStateDictConverter):
                 unet_dict[key] = lora_args
             else:
                 raise ValueError(f"Unsupported key: {key}")
+        # clip skip
+        te1_dict = {k: v for k, v in te1_dict.items() if not k.startswith('encoders.11')}
         return {"unet": unet_dict, "text_encoder": te1_dict, "text_encoder_2": te2_dict}
 
     def convert(self, lora_state_dict: Dict[str, torch.Tensor]) -> Dict[str, Dict[str, torch.Tensor]]:

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/diffsynth_engine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.3.4
+Version: 0.3.6.dev2
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent

{diffsynth_engine-0.3.4 → diffsynth_engine-0.3.6.dev2}/pyproject.toml

@@ -59,6 +59,7 @@ diffsynth_engine = [
 ]
 
 [tool.setuptools_scm]
+local_scheme = "no-local-version"
 
 [tool.ruff]
 # Exclude a variety of commonly ignored directories.