Spaces:

wt3639
/

Course_rec

Paused

Tao Wu commited on Jul 2, 2024

Commit

06e7b29

1 Parent(s): f5b101a

fix bugs

Files changed (1) hide show

app/embedding_setup.py CHANGED Viewed

@@ -106,6 +106,7 @@ def evaluate(
             **kwargs,
         )
         with torch.no_grad():
             generation_output = rec_adapter.generate(
                 **inputs,
                 generation_config=generation_config,

             **kwargs,
         )
         with torch.no_grad():
+            rec_adapter.to(device)
             generation_output = rec_adapter.generate(
                 **inputs,
                 generation_config=generation_config,