Spaces:
Sleeping
Sleeping
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! π’
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:
|