Aria

Sleeping

m-ric HF Staff commited on Dec 9, 2024

Commit

fcea33c

verified ·

1 Parent(s): 650817c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ processor = AutoProcessor.from_pretrained(
 model = AutoModel.from_pretrained("rhymes-ai/Aria",
     revision="f7c92bba",
     torch_dtype=torch.bfloat16,
-    device="auto",
 ).to("cuda")
 @spaces.GPU

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