asigalov61 commited on
Commit
be23228
·
verified ·
1 Parent(s): c21fc79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -416,8 +416,9 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
416
  with ctx:
417
  with torch.no_grad()
418
  out = model(inp)
419
- cache = out[2]
420
- src_embedings = cache.layer_hiddens[-1]
 
421
 
422
  torch.cuda.empty_cache()
423
 
 
416
  with ctx:
417
  with torch.no_grad()
418
  out = model(inp)
419
+
420
+ cache = out[2]
421
+ src_embedings = cache.layer_hiddens[-1]
422
 
423
  torch.cuda.empty_cache()
424