turing-motors
/

heron-chat-blip-ja-stablelm-base-7b-v0

text2text-generation

image-captioning

Model card Files Files and versions Community

tanahhh commited on Sep 7, 2023

Commit

1ca9d23

•

1 Parent(s): 8babac7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ import torch
 from heron.models.video_blip import VideoBlipForConditionalGeneration, VideoBlipProcessor
 from transformers import LlamaTokenizer
-device_id = 1
 device = f"cuda:{device_id}"
 max_length = 512

 from heron.models.video_blip import VideoBlipForConditionalGeneration, VideoBlipProcessor
 from transformers import LlamaTokenizer
+device_id = 0
 device = f"cuda:{device_id}"
 max_length = 512