Skip to content

Commit

Permalink
delete internvl
Browse files Browse the repository at this point in the history
  • Loading branch information
gokayfem committed Nov 3, 2024
1 parent 7817b69 commit e1a48db
Show file tree
Hide file tree
Showing 5 changed files with 1 addition and 125 deletions.
1 change: 0 additions & 1 deletion __init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@ def check_requirements_installed(requirements_path):
"llavaloader",
"suggest",
"joytag",
"internlm",
"uform",
"kosmos2",
"audioldm2",
Expand Down
35 changes: 0 additions & 35 deletions install_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,41 +202,6 @@ def install_llama(system_info):
print(f"CPU installation failed: {e}")
return False

def install_autogptq(system_info):
# Check OS compatibility
imported = package_is_installed("auto_gptq")
if imported:
print("AutoGPTQ installed")
else:
if system_info['os'] not in ['Linux', 'Windows']:
print("AutoGPTQ is not supported on your operating system.")
return

# Prepare base install command
base_command = [sys.executable, "-m", "pip", "install", "auto-gptq"]

# Determine the specific install command based on GPU and CUDA/ROCm version
if system_info['gpu']:
if 'cuda_version' in system_info and system_info['cuda_version'] in ['cu118', 'cu121']:
if system_info['cuda_version'] == 'cu118':
base_command += ["--extra-index-url", "https://huggingface.github.io/autogptq-index/whl/cu118/"]
# No extra URL needed for cu121 as it's the default
elif 'rocm_version' in system_info and system_info['rocm_version'] == 'rocm573':
base_command += ["--extra-index-url", "https://huggingface.github.io/autogptq-index/whl/rocm573/"]
else:
print("Unsupported GPU configuration for AutoGPTQ.")
return
else:
print("No GPU detected. AutoGPTQ installation requires a GPU with CUDA or ROCm support.")
return

# Execute the installation command
try:
print(f"Installing AutoGPTQ with command: {' '.join(base_command)}")
subprocess.check_call(base_command)
except Exception as e:
print(f"Failed to install AutoGPTQ: {e}")

config = None

def is_logging_enabled():
Expand Down
87 changes: 0 additions & 87 deletions nodes/internlm.py

This file was deleted.

2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name = "comfyui_vlm_nodes"
description = "Custom Nodes for Vision Language Models (VLM) , Large Language Models (LLM), Image Captioning, Automatic Prompt Generation, Creative and Consistent Prompt Suggestion, Keyword Extraction"
version = "1.0.5"
license = { file = "LICENSE" }
dependencies = ["accelerate>=0.27.0", "auto-gptq>=0.7.0", "bitsandbytes", "cffi", "decord" , "diffusers" , "diskcache" , "einops>=0.7.0" , "gitpython", "huggingface-hub>=0.20.3", "moviepy", "openai>=0.27.8", "opencv-python", "optimum>=1.17.0", "pillow>=9.4.0", "py-cpuinfo>=3.3.0", "python-dateutil>=2.7.0", "pytz", "qwen-vl-utils", "safetensors>=0.4.1", "scikit-build", "six", "soundfile", "symusic", "torch>=2.0.1,<3.0.0", "torchvision>=0.15.2", "transformers>=4.38.2", "typing"]
dependencies = ["accelerate>=0.27.0", "bitsandbytes", "cffi", "decord" , "diffusers" , "diskcache" , "einops>=0.7.0" , "gitpython", "huggingface-hub>=0.20.3", "moviepy", "openai>=0.27.8", "opencv-python", "optimum>=1.17.0", "pillow>=9.4.0", "py-cpuinfo>=3.3.0", "python-dateutil>=2.7.0", "pytz", "qwen-vl-utils", "safetensors>=0.4.1", "scikit-build", "six", "soundfile", "symusic", "torch>=2.0.1,<3.0.0", "torchvision>=0.15.2", "transformers>=4.38.2", "typing"]

[project.urls]
Repository = "https://github.com/gokayfem/ComfyUI_VLM_nodes"
Expand Down
1 change: 0 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
accelerate>=0.27.0
auto-gptq>=0.7.0
bitsandbytes
cffi
decord
Expand Down

0 comments on commit e1a48db

Please sign in to comment.