jncraton commited on
Commit
558b0e3
1 Parent(s): 4a4b83a

Enable syntax highlighting for example code

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -28,7 +28,8 @@ Let me know!
28
  Phi does not support device_map "auto", and does not seem to want to inference in fp16, so use bf16.
29
 
30
  Here is working code to inference, though it can be improved:
31
- ```
 
32
  import torch
33
  from transformers import AutoModelForCausalLM, AutoTokenizer
34
 
 
28
  Phi does not support device_map "auto", and does not seem to want to inference in fp16, so use bf16.
29
 
30
  Here is working code to inference, though it can be improved:
31
+
32
+ ```python
33
  import torch
34
  from transformers import AutoModelForCausalLM, AutoTokenizer
35