VinayHajare commited on
Commit
182197c
·
verified ·
1 Parent(s): bb883c8

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -29,7 +29,7 @@ clip_model = clip_model.eval()
29
  text_encoder = CLIP_TXT_ENCODER(clip_model).to(device)
30
 
31
  # loading the model from the repository and extracting the generator model
32
- model_path = hf_hub_download(repo_id = repo_id, filename = file_name)
33
  checkpoint = torch.load(model_path, map_location=torch.device(device))
34
  netG = NetG(64, 100, 512, 256, 3, False, clip_model).to(device)
35
  generator = load_model_weights(netG, checkpoint['model']['netG'], multi_gpus=False)
 
29
  text_encoder = CLIP_TXT_ENCODER(clip_model).to(device)
30
 
31
  # loading the model from the repository and extracting the generator model
32
+ model_path = hf_hub_download(repo_id = repo_id, filename = file_name, token = HF_TOKEN)
33
  checkpoint = torch.load(model_path, map_location=torch.device(device))
34
  netG = NetG(64, 100, 512, 256, 3, False, clip_model).to(device)
35
  generator = load_model_weights(netG, checkpoint['model']['netG'], multi_gpus=False)