himanshunitrr commited on
Commit
e3f5c06
1 Parent(s): c5e3bf4

Importing torch

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -48,6 +48,7 @@ Then, follow these steps:
48
  from transformers import AutoModelForCausalLM, AutoProcessor, GenerationConfig
49
  from PIL import Image
50
  import requests
 
51
 
52
  # load the processor
53
  processor = AutoProcessor.from_pretrained(
 
48
  from transformers import AutoModelForCausalLM, AutoProcessor, GenerationConfig
49
  from PIL import Image
50
  import requests
51
+ import torch
52
 
53
  # load the processor
54
  processor = AutoProcessor.from_pretrained(