Update modules.py

TheNeodev · web-flow · commit afc2e7c1d96f · 2025-01-29T14:58:50.000+07:00
diff --git a/lib/modules.py b/lib/modules.py
@@ -59,7 +59,6 @@ def load_hubert():
         suffix="",
     )
     hubert_model = models[0]
-    hubert_model = hubert_model.to(config.device)
     hubert_model = hubert_model.float()
     hubert_model.eval()
 

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,6 @@ def load_hubert():`
`59`	`59`	`suffix="",`
`60`	`60`	`)`
`61`	`61`	`hubert_model = models[0]`
`62`		`- hubert_model = hubert_model.to(config.device)`
`63`	`62`	`hubert_model = hubert_model.float()`
`64`	`63`	`hubert_model.eval()`
`65`	`64`