Chan-Y commited on
Commit
6736065
1 Parent(s): ee1f1f1

fix example code

Browse files

import torch before using it

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -34,6 +34,7 @@ import requests
34
 
35
  from PIL import Image
36
  from transformers import AutoProcessor, AutoModelForCausalLM
 
37
 
38
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
39
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
34
 
35
  from PIL import Image
36
  from transformers import AutoProcessor, AutoModelForCausalLM
37
+ import torch
38
 
39
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
40
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32