Update README.md

#1
by dyier - opened
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -22,6 +22,7 @@ git clone https://github.com/X-PLUG/mPLUG-Owl.git
22
  from transformers import AutoTokenizer
23
  from mplug_owl.modeling_mplug_owl import MplugOwlForConditionalGeneration
24
  from mplug_owl.processing_mplug_owl import MplugOwlImageProcessor, MplugOwlProcessor
 
25
 
26
  pretrained_ckpt = 'MAGAer13/mplug-owl-bloomz-7b-multilingual'
27
  model = MplugOwlForConditionalGeneration.from_pretrained(
 
22
  from transformers import AutoTokenizer
23
  from mplug_owl.modeling_mplug_owl import MplugOwlForConditionalGeneration
24
  from mplug_owl.processing_mplug_owl import MplugOwlImageProcessor, MplugOwlProcessor
25
+ import torch
26
 
27
  pretrained_ckpt = 'MAGAer13/mplug-owl-bloomz-7b-multilingual'
28
  model = MplugOwlForConditionalGeneration.from_pretrained(