Spaces:
Bradarr
/
Running on Zero

Bradarr commited on
Commit
a7d7d4e
·
verified ·
1 Parent(s): 7f2eda6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -107,11 +107,11 @@ def infer(user_audio) -> tuple[int, np.ndarray]:
107
  generator = load_csm_1b(model_path, device)
108
  logging.info("Sesame CSM 1B loaded successfully.")
109
 
110
- whisper_model = whisper.load_model("small.en", device=device, dtype=torch.bfloat16)
111
  logging.info("Whisper model loaded successfully.")
112
 
113
  tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
114
- model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it").to(device, dtype=torch.bfloat16)
115
  logging.info("Gemma 3 1B pt model loaded successfully.")
116
 
117
  if not user_audio:
 
107
  generator = load_csm_1b(model_path, device)
108
  logging.info("Sesame CSM 1B loaded successfully.")
109
 
110
+ whisper_model = whisper.load_model("small.en", device=device)
111
  logging.info("Whisper model loaded successfully.")
112
 
113
  tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
114
+ model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it").to(device)
115
  logging.info("Gemma 3 1B pt model loaded successfully.")
116
 
117
  if not user_audio: