diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py
index 7f991d6..32bfbda 100644
--- a/diffusers_in_comfy.py
+++ b/diffusers_in_comfy.py
@@ -494,7 +494,7 @@ class LoRALoader:
 
     def load_lora(self, pipeline, lora_name, lora_scale):
         lora_path = folder_paths.get_full_path("loras", lora_name)
-        lora_name = os.path.splitext(lora_name)[0]
+        lora_name = os.path.splitext(lora_name)[0] # to remove the extension from the name
         pipeline.load_lora_weights(lora_path, adapter_name=lora_name)
         pipeline.set_adapters([lora_name], adapter_weights=[lora_scale])
 
diff --git a/pyproject.toml b/pyproject.toml
index f0c724d..d05112b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "diffusers-in-comfyui"
 description = "A collection of ComfyUI custom nodes that allow to use most Diffusers pipelines and components in Comfy(such as Txt2Img, Img2Img, Inpainting, LoRAS, B-LoRAS, ControlNet...). You will find them under 'Diffusers-in-Comfy'!"
-version = "1.0.0"
+version = "1.0.1"
 license = {file = "LICENSE"}
 dependencies = ["absl-py==2.1.0", "accelerate==0.33.0", "aiohttp==3.9.5", "aiosignal==1.3.1", "antlr4-python3-runtime==4.9.3", "async-timeout==4.0.3", "attrs==23.2.0", "bitsandbytes==0.43.1", "certifi==2024.7.4", "cffi==1.16.0", "charset-normalizer==3.3.2", "colorama==0.4.6", "diffusers @ git+https://github.com/huggingface/diffusers@3bb1fd6fc05917ce2fc577fcda2163391c5bdcc0", "einops==0.8.0", "filelock==3.15.4", "frozenlist==1.4.1", "fsspec==2024.6.1", "ftfy==6.2.0", "grpcio==1.65.1", "huggingface-hub==0.24.1", "idna==3.7", "importlib_metadata==8.1.0", "intel-openmp==2021.4.0", "Jinja2==3.1.4", "kornia==0.7.3", "kornia_rs==0.1.5", "Markdown==3.6", "MarkupSafe==2.1.5", "mkl==2021.4.0", "mpmath==1.3.0", "multidict==6.0.5", "networkx==3.2.1", "numpy==1.26.4", "omegaconf==2.3.0", "opencv-python==4.10.0.84", "packaging==24.1", "peft==0.11.1", "pillow==10.4.0", "protobuf==4.25.3", "psutil==6.0.0", "pycparser==2.22", "PyYAML==6.0.1", "regex==2024.5.15", "requests==2.32.3", "safetensors==0.4.3", "scipy==1.13.1", "sentencepiece==0.2.0", "six==1.16.0", "soundfile==0.12.1", "spandrel==0.3.4", "sympy==1.13.1", "tbb==2021.13.0", "tensorboard==2.17.0", "tensorboard-data-server==0.7.2", "tokenizers==0.19.1", "torch==2.3.1+cu121", "torchaudio==2.3.1+cu121", "torchsde==0.2.6", "torchvision==0.18.1", "tqdm==4.66.4", "trampoline==0.1.2", "transformers==4.43.0", "typing_extensions==4.12.2", "urllib3==2.2.2", "wcwidth==0.2.13", "Werkzeug==3.0.3", "xformers==0.0.27", "yarl==1.9.4", "zipp==3.19.2"]
 
diff --git a/utils.py b/utils.py
index 122041f..6fdb4ea 100644
--- a/utils.py
+++ b/utils.py
@@ -12,10 +12,11 @@ import folder_paths
 from PIL import Image, ImageOps
 from abc import ABC, abstractmethod
 
-class ImageInference:
+class ImageInference(ABC):
     def setup_generator(self, seed):
         return torch.Generator(device='cuda').manual_seed(seed)
 
+    @abstractmethod
     def infer_image(self, **kwargs):
         raise NotImplementedError