Update README.md
#1
by
dyier
- opened
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(
|