Spaces:
Runtime error
Runtime error
'adjustments'
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ import torch
|
|
23 |
import torch.optim
|
24 |
import torch.nn as nn
|
25 |
from torch.nn import functional as F
|
|
|
26 |
import torchvision.transforms.functional as TF
|
27 |
import torchvision.transforms as T
|
28 |
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}
|
|
39 |
|
40 |
clip_normalize = T.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], std=[0.26862954, 0.26130258, 0.27577711])
|
41 |
clip_model = 'ViT-B/16'
|
42 |
-
sideX, sideY =
|
43 |
inv_color_scale = 1.6
|
44 |
anneal_lr = True
|
45 |
display_augs = False
|
|
|
23 |
import torch.optim
|
24 |
import torch.nn as nn
|
25 |
from torch.nn import functional as F
|
26 |
+
import torchvision
|
27 |
import torchvision.transforms.functional as TF
|
28 |
import torchvision.transforms as T
|
29 |
import numpy as np
|
|
|
40 |
|
41 |
clip_normalize = T.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], std=[0.26862954, 0.26130258, 0.27577711])
|
42 |
clip_model = 'ViT-B/16'
|
43 |
+
sideX, sideY = 256, 256 # Resolution
|
44 |
inv_color_scale = 1.6
|
45 |
anneal_lr = True
|
46 |
display_augs = False
|