mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix some imports..

Browse Source
This commit is contained in:
James Betker 2022-07-28 02:35:32 -06:00
parent cfe907f13f
commit 1d68624828
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/models/image_latents/vit_latent.py
View File

@ -2,9 +2,9 @@ import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from codes.models.arch_util import ResBlock from models.arch_util import ResBlock
from codes.models.lucidrains.x_transformers import Encoder from models.lucidrains.x_transformers import Encoder
from codes.trainer.networks import register_model from trainer.networks import register_model
class VitLatent(nn.Module): class VitLatent(nn.Module):