D:\Anaconda\envs\viscpm\lib\site-packages\torchvision\models\detection\anchor_utils.py:63: UserWarning: Failed to initialize NumPy: module compiled against API version 0x10 but this version of numpy is 0xf (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:77.)
device: torch.device = torch.device("cpu"),
Traceback (most recent call last):
File "D:\ViscpmProj\VisCPM\hello.py", line 5, in
viscpm_chat = VisCPMChat(model_path, image_safety_checker=True)
File "D:\ViscpmProj\VisCPM\VisCPM\viscpm_chat.py", line 40, in init
self.cpm_model = CPMBeeTorch(self.config)
File "D:\ViscpmProj\VisCPM\VisCPM\models\cpmbee.py", line 74, in init
self.input_embedding = EmbeddingExt(
File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\embedding.py", line 77, in init
self.rotary_emb = RotaryEmbedding(
File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\position_embedding.py", line 222, in init
base ** (torch.arange(0, dim, 2, device="cuda", dtype=torch.float32) / dim)
File "D:\Anaconda\envs\viscpm\lib\site-packages\torch\cuda__init__.py", line 221, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
D:\Anaconda\envs\viscpm\lib\site-packages\torchvision\models\detection\anchor_utils.py:63: UserWarning: Failed to initialize NumPy: module compiled against API version 0x10 but this version of numpy is 0xf (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:77.) device: torch.device = torch.device("cpu"), Traceback (most recent call last): File "D:\ViscpmProj\VisCPM\hello.py", line 5, in
viscpm_chat = VisCPMChat(model_path, image_safety_checker=True)
File "D:\ViscpmProj\VisCPM\VisCPM\viscpm_chat.py", line 40, in init
self.cpm_model = CPMBeeTorch(self.config)
File "D:\ViscpmProj\VisCPM\VisCPM\models\cpmbee.py", line 74, in init
self.input_embedding = EmbeddingExt(
File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\embedding.py", line 77, in init
self.rotary_emb = RotaryEmbedding(
File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\position_embedding.py", line 222, in init
base ** (torch.arange(0, dim, 2, device="cuda", dtype=torch.float32) / dim)
File "D:\Anaconda\envs\viscpm\lib\site-packages\torch\cuda__init__.py", line 221, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled