VinayHajare commited on
Commit
fb5d9d7
·
verified ·
1 Parent(s): 9af7efd

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -20,7 +20,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
20
 
21
  # repositiory of the model
22
  repo_id = "VinayHajare/EfficientCLIP-GAN"
23
- file_name = "saved_models/state_epoch_120.pth"
24
 
25
  # clip model wrapped with the custom encoder
26
  clip_text = "ViT-B/32"
 
20
 
21
  # repositiory of the model
22
  repo_id = "VinayHajare/EfficientCLIP-GAN"
23
+ file_name = "saved_models/state_epoch_1480.pth"
24
 
25
  # clip model wrapped with the custom encoder
26
  clip_text = "ViT-B/32"