Hugging Face's logo

Spaces:
tddandroid
/
wan2-2-fp8da-aoti-faster
Runtime error

App Files Community
tddandroid commited on
Commit
ed66a8c
·
verified ·
1 Parent(s): f4d9e3e

Create get_device

Browse files
Files changed (1) hide show
  1. get_device +4 -0
get_device ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from setup_device import get_device
2
+
3
+ device = get_device()
4
+ model.to(device)