Spaces:

OptimalScale
/

Robin-7b

Runtime error

hendrydong commited on Apr 23, 2023

Commit

74fa773

1 Parent(s): e4f3acf

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ def main():
         model_args,
         tune_strategy='none',
         ds_config=ds_config,
-        device="cpu",#pipeline_args.device,
     )
     # We don't need input data, we will read interactively from stdin

         model_args,
         tune_strategy='none',
         ds_config=ds_config,
+        device=pipeline_args.device,
     )
     # We don't need input data, we will read interactively from stdin