Add normalisation

app.py

@@ -4,6 +4,7 @@ import sys
 
 from IPython import display
 import torch
+from torchvision import transforms
 from torchvision import utils as tv_utils
 from torchvision.transforms import functional as TF
 import gradio as gr
@@ -28,7 +29,10 @@ _, side_y, side_x = model.shape
 model.load_state_dict(torch.load(cc12m_model, map_location='cpu'))
 model = model.half().cuda().eval().requires_grad_(False)
 clip_model = clip.load(model.clip_model, jit=False, device='cpu')[0]
-
+clip_model.eval().requires_grad_(False)
+normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],
+                                     std=[0.26862954, 0.26130258, 0.27577711])
+make_cutouts = MakeCutouts(clip_model.visual.input_resolution, 16, 1.)
 def run_all(prompt, steps, n_images, weight, clip_guided):
     import random
     seed = int(random.randint(0, 2147483647))