'adjustments'

app.py

@@ -23,6 +23,7 @@ import torch
 import torch.optim
 import torch.nn as nn
 from torch.nn import functional as F
+import torchvision
 import torchvision.transforms.functional as TF
 import torchvision.transforms as T
 import numpy as np
@@ -39,7 +40,7 @@ clip_models = {'ViT-B/32': clip_model_vit_b_32, 'ViT-B/16': clip_model_vit_b_16}
 
 clip_normalize = T.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], std=[0.26862954, 0.26130258, 0.27577711])
 clip_model = 'ViT-B/16'
-sideX, sideY = 512, 512  # Resolution
+sideX, sideY = 256, 256  # Resolution
 inv_color_scale = 1.6
 anneal_lr = True
 display_augs = False