Skip to content

Commit 9393b17

Browse files
authored
Add warning in PTEModel when not defined
1 parent 2b555b1 commit 9393b17

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchchat/model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1062,5 +1062,6 @@ def forward(self, x, input_pos):
10621062
def setup_caches(self, max_batch_size, max_seq_length):
10631063
pass
10641064

1065-
except:
1065+
except Exception as e:
1066+
print(f"Warning: PTEModel (ExecuTorch not available with exception: {e}")
10661067
pass

0 commit comments

Comments
 (0)