fbnnb commited on
Commit
0873485
1 Parent(s): 329a809

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -144,6 +144,10 @@ def get_image(image, sketch, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, see
144
  transforms.Resize(min(resolution)),
145
  transforms.CenterCrop(resolution),
146
  ])
 
 
 
 
147
  print("before empty cache")
148
  torch.cuda.empty_cache()
149
  print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
 
144
  transforms.Resize(min(resolution)),
145
  transforms.CenterCrop(resolution),
146
  ])
147
+
148
+ transform = transforms.Compose([
149
+ transforms.Resize(resolution),
150
+ ])
151
  print("before empty cache")
152
  torch.cuda.empty_cache()
153
  print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))