Ahsen Khaliq commited on
Commit
1d40feb
Β·
1 Parent(s): 8828289

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -198,8 +198,8 @@ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
198
  print('Using device:', device)
199
  model = load_vqgan_model(args.vqgan_config, args.vqgan_checkpoint).to(device)
200
  perceptor = clip.load(args.clip_model, jit=False)[0].eval().requires_grad_(False).to(device)
201
- all_frames = []
202
  def inference(text, seed, step_size, max_iterations, width, height, init_image, init_weight):
 
203
  size=[width, height]
204
  texts = text
205
  init_weight=init_weight
 
198
  print('Using device:', device)
199
  model = load_vqgan_model(args.vqgan_config, args.vqgan_checkpoint).to(device)
200
  perceptor = clip.load(args.clip_model, jit=False)[0].eval().requires_grad_(False).to(device)
 
201
  def inference(text, seed, step_size, max_iterations, width, height, init_image, init_weight):
202
+ all_frames = []
203
  size=[width, height]
204
  texts = text
205
  init_weight=init_weight