izhx commited on
Commit
333d6e0
·
verified ·
1 Parent(s): bd36ab0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -3697,6 +3697,8 @@ The `GME` models support three types of input: **text**, **image**, and **image-
3697
  **Transformers**
3698
 
3699
  ```python
 
 
3700
  t2i_prompt = 'Find an image that matches the given text.'
3701
  texts = [
3702
  "The Tesla Cybertruck is a battery electric pickup truck built by Tesla, Inc. since 2023.",
 
3697
  **Transformers**
3698
 
3699
  ```python
3700
+ from transformers import AutoModel
3701
+
3702
  t2i_prompt = 'Find an image that matches the given text.'
3703
  texts = [
3704
  "The Tesla Cybertruck is a battery electric pickup truck built by Tesla, Inc. since 2023.",