File tree Expand file tree Collapse file tree 3 files changed +3
-12
lines changed
taggui/auto_captioning/models Expand file tree Collapse file tree 3 files changed +3
-12
lines changed Original file line number Diff line number Diff line change @@ -34,8 +34,3 @@ https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp311-cp311-win_amd64
34
34
flash-attn == 2.7.4.post1 ; platform_system == "Linux"
35
35
https://github.com/kingbri1/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu124torch2.6.0cxx11abiFALSE-cp312-cp312-win_amd64.whl; platform_system == "Windows" and python_version == "3.12"
36
36
https://github.com/kingbri1/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu124torch2.6.0cxx11abiFALSE-cp311-cp311-win_amd64.whl; platform_system == "Windows" and python_version == "3.11"
37
-
38
- # Triton (CogVLM2)
39
- triton == 3.2.0 ; platform_system == "Linux"
40
- https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp312-cp312-win_amd64.whl; platform_system == "Windows" and python_version == "3.12"
41
- https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp311-cp311-win_amd64.whl; platform_system == "Windows" and python_version == "3.11"
Original file line number Diff line number Diff line change 1
1
# -*- mode: python ; coding: utf-8 -*-
2
- import os , sys
3
2
from PyInstaller .utils .hooks import collect_data_files
4
3
5
4
datas = [('clip-vit-base-patch32' , 'clip-vit-base-patch32' ),
6
5
('images/icon.ico' , 'images' )]
7
- datas += [(os .path .join (sys .base_prefix , 'include' ), 'include' )]
8
- datas += collect_data_files ('triton' )
9
6
datas += collect_data_files ('xformers' )
10
7
hiddenimports = [
11
8
'timm.models.layers' ,
@@ -30,7 +27,6 @@ a = Analysis(
30
27
cipher = block_cipher ,
31
28
noarchive = False ,
32
29
module_collection_mode = {
33
- 'triton' : 'py' ,
34
30
'xformers' : 'pyz+py' ,
35
31
},
36
32
)
Original file line number Diff line number Diff line change @@ -18,9 +18,9 @@ class Cogvlm2(AutoCaptioningModel):
18
18
19
19
def get_additional_error_message (self ) -> str | None :
20
20
if not importlib .util .find_spec ('triton' ):
21
- return ('This model requires the `triton` package, which is only '
22
- 'available for Linux and Windows. Therefore, this model '
23
- 'cannot be run on this system.' )
21
+ return ('This model cannot be run because it requires the '
22
+ '`triton` package, which is not available for your '
23
+ 'system.' )
24
24
is_4_bit_model = 'int4' in self .model_id
25
25
if is_4_bit_model :
26
26
if self .device_setting == CaptionDevice .CPU :
You can’t perform that action at this time.
0 commit comments