Hugging Face's logo

Spaces:
EpistemeAI
/
AudioGemma
Paused

App Files Community
legolasyiu commited on
Commit
80fb53a
·
verified ·
1 Parent(s): b51b3fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ from transformers import AutoProcessor, AutoModelForImageTextToText
8
  processor = AutoProcessor.from_pretrained("EpistemeAI/Audiogemma-3N-finetune")
9
  model = AutoModelForImageTextToText.from_pretrained(
10
  "EpistemeAI/Audiogemma-3N-finetune",
11
- device="cuda",
12
  torch_dtype=torch.bfloat16,
13
- )
 
14
 
15
 
16
 
 
8
  processor = AutoProcessor.from_pretrained("EpistemeAI/Audiogemma-3N-finetune")
9
  model = AutoModelForImageTextToText.from_pretrained(
10
  "EpistemeAI/Audiogemma-3N-finetune",
 
11
  torch_dtype=torch.bfloat16,
12
+ ).to("cuda")
13
+
14
 
15
 
16