Aria

Running on Zero

m-ric HF Staff commited on Dec 9, 2024

Commit

7a3273b

verified ·

1 Parent(s): fcea33c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model = AutoModel.from_pretrained("rhymes-ai/Aria",
     revision="f7c92bba",
     torch_dtype=torch.bfloat16,
     device_map="auto",
-).to("cuda")
 @spaces.GPU
 def model_inference(

     revision="f7c92bba",
     torch_dtype=torch.bfloat16,
     device_map="auto",
+)
 @spaces.GPU
 def model_inference(