Update hf-inference-example.py

3 years ago · 0eb2a070ea
parent ae518062fa
commit 0eb2a070ea
1 changed files with 1 additions and 1 deletions
--- a/hf-inference-example.py
+++ b/hf-inference-example.py
@ -1,7 +1,7 @@
 import llamahf
 import os

-# # to save memory use bfloat16 on cpu
+# # to save memory use bfloat16
 # import torch
 # torch.set_default_dtype(torch.bfloat16)