Skip to content

Commit

Permalink
Upgrade force int8 message to warning log level
Browse files Browse the repository at this point in the history
  • Loading branch information
kristiankielhofner committed Apr 10, 2023
1 parent 31ae203 commit 4e32596
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ async def create_datagram_endpoint(self, protocol_factory,

# Override compute_type if at least one non-Turing card
if cuda_device_capability <= 70:
logger.info(f'CUDA: Device {cuda_dev_num} is pre-Turing, forcing int8')
logger.warning(f'CUDA: Device {cuda_dev_num} is pre-Turing, forcing int8')
compute_type = "int8"

# Set ctranslate device index based on number of supported devices
Expand Down

0 comments on commit 4e32596

Please sign in to comment.