asigalov61
commited on
Update app.py
Browse files
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 |
-
|
420 |
-
|
|
|
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 |
|