Spaces:
Runtime error
Runtime error
VinayHajare
commited on
Update inference.py
Browse files- 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)
|