luodian commited on
Commit
bac706e
1 Parent(s): ae1367b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -1,3 +1,10 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ ```python
6
+ from flamingo.modeling_flamingo import FlamingoForConditionalGeneration
7
+ flamingo_model = FlamingoForConditionalGeneration.from_pretrained("luodian/Flamingo-Llama2-Chat7B-CC3M", device_map=auto)
8
+ prompt = "<image>an image of"
9
+ simple_prompt = "<image>"
10
+ ```