Jyothirmai commited on
Commit
fe70bc2
1 Parent(s): a86795b

Update clipGPT.py

Browse files
Files changed (1) hide show
  1. clipGPT.py +6 -3
clipGPT.py CHANGED
@@ -2,14 +2,17 @@ from transformers import AutoTokenizer, AutoModel
2
  import clip
3
  import skimage.io as io
4
  import PIL.Image
5
-
 
 
6
  import torch
7
  import torch.nn as nn
8
  from torch.utils.data import Dataset, DataLoader
9
  from torch.nn import functional as F
10
- from typing import Tuple
11
-
12
  import pandas as pd
 
 
 
13
  import numpy as np
14
  import time
15
  import json
 
2
  import clip
3
  import skimage.io as io
4
  import PIL.Image
5
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel, AdamW, get_linear_schedule_with_warmup
6
+ import os
7
+ import pickle
8
  import torch
9
  import torch.nn as nn
10
  from torch.utils.data import Dataset, DataLoader
11
  from torch.nn import functional as F
 
 
12
  import pandas as pd
13
+ from tqdm import tqdm
14
+ from PIL import Image
15
+ from typing import Tuple
16
  import numpy as np
17
  import time
18
  import json