Merge pull request #24 from TheFutureGadgetsLab/main
Fixed disabling of gradients in the torch code
This commit is contained in:
@@ -2,7 +2,7 @@ import numpy
|
||||
from typing import Dict
|
||||
from torch import LongTensor, FloatTensor
|
||||
import torch
|
||||
torch.no_grad()
|
||||
torch.set_grad_enabled(False)
|
||||
|
||||
from .models.vqgan_detokenizer import VQGanDetokenizer
|
||||
from .models.dalle_bart_encoder_torch import DalleBartEncoderTorch
|
||||
|
Reference in New Issue
Block a user