JonasGeiping commited on
Commit
288f483
·
verified ·
1 Parent(s): bb6b474

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -26,7 +26,8 @@ All details on this model can be found in the tech report: "Scaling up Test-Time
26
 
27
  ## Downloading and Using the Model
28
  Load the model like this:
29
- ```python
 
30
  from transformers import AutoModelForCausalLM, AutoTokenizer
31
 
32
  model = AutoModelForCausalLM.from_pretrained("tomg-group-umd/huginn-0125", torch_dtype=torch.bfloat16, trust_remote_code=True)
 
26
 
27
  ## Downloading and Using the Model
28
  Load the model like this:
29
+ ```python
30
+ import torch
31
  from transformers import AutoModelForCausalLM, AutoTokenizer
32
 
33
  model = AutoModelForCausalLM.from_pretrained("tomg-group-umd/huginn-0125", torch_dtype=torch.bfloat16, trust_remote_code=True)