license: mit | |
```python | |
from flamingo.modeling_flamingo import FlamingoForConditionalGeneration | |
flamingo_model = FlamingoForConditionalGeneration.from_pretrained("luodian/Flamingo-Llama2-Chat7B-CC3M", device_map=auto) | |
prompt = "<image>an image of" | |
simple_prompt = "<image>" | |
``` |