Merge pull request #31 from VE-FORBRYDERNE/cpu

Fix gen_in device logic in generate()
2021-11-17 22:42:31 +01:00 · 2021-11-17 22:42:31 +01:00 · 27ee45b9cc
parent e71271933a 2f0b673b28
commit 27ee45b9cc
1 changed files with 0 additions and 2 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -1458,8 +1458,6 @@ def generate(txt, minimum, maximum, found_entries=None):
            gen_in = gen_in.to(0)
        elif(vars.hascuda and vars.breakmodel):
            gen_in = gen_in.to(breakmodel.primary_device)
-        elif(vars.hascuda):
-            gen_in = gen_in.to(0)
        else:
            gen_in = gen_in.to('cpu')