From 40aa304b7e391386fc05ee8a574fdddbe1a94bde Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 12 Apr 2022 11:45:57 -0700
Subject: [PATCH] rename to DiffusionPriorNetwork in case ARPriorNetwork is
 ever built

---
 dalle2_pytorch/dalle2_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index 5712c6a..89d64e9 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -149,7 +149,7 @@ class Transformer(nn.Module):
 
         return self.norm(x)
 
-class PriorNetwork(nn.Module):
+class DiffusionPriorNetwork(nn.Module):
     def __init__(
         self,
         dim,