Gemma-4-Multi

Running on Zero

SeaWolf-AI commited on 1 day ago

Commit

59d32ff

verified ·

1 Parent(s): daa968b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ def _load_model(model_name: str):
         import gc; gc.collect()
         print(f"[MODEL] Unloaded previous model", flush=True)
-    _processor = AutoProcessor.from_pretrained(model_id, backend="pil")
     _model = AutoModelForMultimodalLM.from_pretrained(
         model_id, device_map="auto", dtype=torch.bfloat16,
     )

         import gc; gc.collect()
         print(f"[MODEL] Unloaded previous model", flush=True)
+    _processor = AutoProcessor.from_pretrained(model_id)
     _model = AutoModelForMultimodalLM.from_pretrained(
         model_id, device_map="auto", dtype=torch.bfloat16,
     )