ThreadAbort commited on
Commit
459b003
Β·
1 Parent(s): 4010472

[Update]: Introduced GPU decorator for memory reconstruction function πŸš€

Browse files

- Added: @spaces.GPU decorator to the generate_memory_reconstruction method to enhance performance with GPU acceleration.
- Pro Tip of the Commit: When in doubt, let the GPU do the heavy lifting! πŸ’ͺ✨
Aye, Aye! 🚒

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -460,7 +460,7 @@ class MemoryWave:
460
  "arousal": self.emotion.arousal.item(),
461
  }
462
  }
463
-
464
  def generate_memory_reconstruction(self,
465
  emotion_valence: float,
466
  corruption_level: float = 0.3) -> Dict:
 
460
  "arousal": self.emotion.arousal.item(),
461
  }
462
  }
463
+ @spaces.GPU(duration=10)
464
  def generate_memory_reconstruction(self,
465
  emotion_valence: float,
466
  corruption_level: float = 0.3) -> Dict: