Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -349,7 +349,7 @@ Speaker 2: ...
|
|
349 |
)
|
350 |
|
351 |
# Move to correct device
|
352 |
-
|
353 |
add_log(f"✅ Inputs moved to device: {self.model.device}")
|
354 |
|
355 |
add_log("self🧠 Generating with model...")
|
|
|
349 |
)
|
350 |
|
351 |
# Move to correct device
|
352 |
+
inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
|
353 |
add_log(f"✅ Inputs moved to device: {self.model.device}")
|
354 |
|
355 |
add_log("self🧠 Generating with model...")
|