removing uv.lock

Author: Ubuntu

pyproject.toml

@@ -70,6 +70,13 @@ name = "pruna_internal"
 url = "https://prunaai.pythonanywhere.com/simple/"  # Pruna Pythonanywhere
 default = true # default = True makes this index the lowest prio
 
+[[tool.uv.index]]
+name = "intel-pytorch-extension"
+url = "https://pytorch-extension.intel.com/release-whl/stable/cpu/cn/"
+
+[tool.uv]
+index-strategy = "unsafe-best-match"
+
 [tool.uv.sources]
 gptqmodel = [
   { index = "pruna_internal", marker = "sys_platform != 'darwin' or platform_machine != 'arm64'"},
@@ -180,12 +187,15 @@ dev = [
     "pytest-rerunfailures",
     "coverage",
     "docutils",
-    "ty==0.0.1a21",
+    "ty",
     "types-PyYAML",
     "logbar",
     "pytest-xdist>=3.8.0",
 ]
 cpu = []
+intel = [
+    "intel-extension-for-pytorch==2.7.0",
+]
 
 [build-system]
 requires = ["hatchling"]

src/pruna/algorithms/x_fast.py

@@ -46,16 +46,7 @@ class XFast(PrunaAlgorithmBase):
     dataset_required: bool = False
     compatible_before: list[str | AlgorithmTag] = [
         "quanto",
-        "half",
-        "text_to_text_lora",
-        "text_to_image_lora",
-        "text_to_image_perp",
-        "text_to_image_inplace_perp",
-        "text_to_text_perp",
-        "text_to_text_inplace_perp",
-        "text_to_image_distillation_inplace_perp",
-        "text_to_image_distillation_lora",
-        "text_to_image_distillation_perp",
+        "half"
     ]
     required_install: str = "``pip install pruna[stable-fast]``"