Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ from loguru import logger
|
|
| 14 |
from PIL import Image
|
| 15 |
from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
|
| 16 |
|
| 17 |
-
model_id = os.getenv("MODEL_ID", "google/gemma-3-
|
| 18 |
processor = AutoProcessor.from_pretrained(model_id, padding_side="left")
|
| 19 |
model = Gemma3ForConditionalGeneration.from_pretrained(
|
| 20 |
model_id, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager"
|
|
|
|
| 14 |
from PIL import Image
|
| 15 |
from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
|
| 16 |
|
| 17 |
+
model_id = os.getenv("MODEL_ID", "google/gemma-3-27b-it")
|
| 18 |
processor = AutoProcessor.from_pretrained(model_id, padding_side="left")
|
| 19 |
model = Gemma3ForConditionalGeneration.from_pretrained(
|
| 20 |
model_id, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager"
|