Hugging Face's logo

Spaces:
ibrim
/
CLIPModel
Runtime error

App Files Community
ibrim commited on
Commit
6fde60c
·
verified ·
1 Parent(s): 66da508

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -4
config.py CHANGED
@@ -1,8 +1,8 @@
1
  import torch
2
 
3
  debug = True
4
- image_path = "/raid/users/mohammadibrahim-st/TSAI/OpenAI-CLIP/Flicker-8k/Images"
5
- captions_path = "/raid/users/mohammadibrahim-st/TSAI/OpenAI-CLIP/Flicker-8k"
6
  batch_size = 20
7
  num_workers = 0
8
  lr = 1e-3
@@ -14,9 +14,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
 
15
  model_name = 'resnet50'
16
  image_embedding = 2048
17
- text_encoder_model = "/raid/users/mohammadibrahim-st/Models/BertDistil"
18
  text_embedding = 768
19
- text_tokenizer = "/raid/users/mohammadibrahim-st/Models/BertDistil"
20
  max_length = 200
21
 
22
  pretrained = False # for both image encoder and text encoder
 
1
  import torch
2
 
3
  debug = True
4
+ image_path = ""
5
+ captions_path = os.getcwd()
6
  batch_size = 20
7
  num_workers = 0
8
  lr = 1e-3
 
14
 
15
  model_name = 'resnet50'
16
  image_embedding = 2048
17
+ text_encoder_model = "distilbert/distilbert-base-uncased"
18
  text_embedding = 768
19
+ text_tokenizer = "distilbert/distilbert-base-uncased"
20
  max_length = 200
21
 
22
  pretrained = False # for both image encoder and text encoder