Spaces:
Runtime error
Runtime error
Fixed model
Browse files
app.py
CHANGED
|
@@ -126,6 +126,7 @@ quant_config = BitsAndBytesConfig(
|
|
| 126 |
)
|
| 127 |
|
| 128 |
# Load Gemma 2B (Quantized) with config
|
|
|
|
| 129 |
model = AutoModelForCausalLM.from_pretrained(
|
| 130 |
model_name,
|
| 131 |
device_map="auto",
|
|
|
|
| 126 |
)
|
| 127 |
|
| 128 |
# Load Gemma 2B (Quantized) with config
|
| 129 |
+
model_name = "google/gemma-2b-it"
|
| 130 |
model = AutoModelForCausalLM.from_pretrained(
|
| 131 |
model_name,
|
| 132 |
device_map="auto",
|