microsoft
/

Florence-2-base

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

fix example code

#21

by Chan-Y - opened about 16 hours ago

base: refs/heads/main

←

from: refs/pr/21

Discussion Files changed

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -34,6 +34,7 @@ import requests
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32

 from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
+import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32