diff --git a/torchchat/model.py b/torchchat/model.py index f50d2a8be..28429370c 100644 --- a/torchchat/model.py +++ b/torchchat/model.py @@ -1062,5 +1062,6 @@ def forward(self, x, input_pos): def setup_caches(self, max_batch_size, max_seq_length): pass -except: +except Exception as e: + print(f"Warning: PTEModel (ExecuTorch) not available with exception: {e}") pass