tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #24 from TheFutureGadgetsLab/main

Browse Source 
Fixed disabling of gradients in the torch code
This commit is contained in:
kuprel 2022-06-28 21:10:35 -04:00 committed by GitHub
commit b8c4173181
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
min_dalle/min_dalle_torch.py
View File

@ -2,7 +2,7 @@ import numpy
from typing import Dict from typing import Dict
from torch import LongTensor, FloatTensor from torch import LongTensor, FloatTensor
import torch import torch
torch.no_grad() torch.set_grad_enabled(False)
from .models.vqgan_detokenizer import VQGanDetokenizer from .models.vqgan_detokenizer import VQGanDetokenizer
from .models.dalle_bart_encoder_torch import DalleBartEncoderTorch from .models.dalle_bart_encoder_torch import DalleBartEncoderTorch