Spaces:

Stevross
/

Astrid-Assistant

Runtime error

Stevross commited on Nov 3, 2023

Commit

24426ff

1 Parent(s): 838ec94

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ generate_text = pipeline(
     model="Stevross/Astrid-3B",
     task="text-generation",
     trust_remote_code=True,
-    device=0  # Use GPU if available, otherwise CPU
 )
 def generate_response(prompt):

     model="Stevross/Astrid-3B",
     task="text-generation",
     trust_remote_code=True,
+    device=1  # Use GPU if available, otherwise CPU
 )
 def generate_response(prompt):