Hugging Face's logo

openai
/
imagegpt-small

Transformers
PyTorch
imagegpt
vision
Model card Files
xet
 Community
2
nielsr HF Staff commited on
Commit
c05f236
·
1 Parent(s): 0e11e14

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,12 +31,12 @@ You can use the raw model for either feature extractor or (un) conditional image
31
  Here is how to use this model in PyTorch to perform unconditional image generation:
32
 
33
  ```python
34
- from transformers import ImageGPTFeatureExtractor, ImageGPTForCausalImageModeling
35
  import torch
36
  import matplotlib.pyplot as plt
37
  import numpy as np
38
 
39
- feature_extractor = ImageGPTFeatureExtractor.from_pretrained('openai/imagegpt-small')
40
  model = ImageGPTForCausalImageModeling.from_pretrained('openai/imagegpt-small')
41
 
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
31
  Here is how to use this model in PyTorch to perform unconditional image generation:
32
 
33
  ```python
34
+ from transformers import ImageGPTImageProcessor, ImageGPTForCausalImageModeling
35
  import torch
36
  import matplotlib.pyplot as plt
37
  import numpy as np
38
 
39
+ feature_extractor = ImageGPTImageProcessor.from_pretrained('openai/imagegpt-small')
40
  model = ImageGPTForCausalImageModeling.from_pretrained('openai/imagegpt-small')
41
 
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")