Spaces:
Runtime error
Runtime error
Fixed
Browse files
app.py
CHANGED
|
@@ -133,6 +133,9 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 133 |
quantization_config=quant_config
|
| 134 |
)
|
| 135 |
|
|
|
|
|
|
|
|
|
|
| 136 |
# Create LLM Pipeline
|
| 137 |
text_gen_pipeline = pipeline(
|
| 138 |
"text-generation",
|
|
|
|
| 133 |
quantization_config=quant_config
|
| 134 |
)
|
| 135 |
|
| 136 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 137 |
+
|
| 138 |
+
|
| 139 |
# Create LLM Pipeline
|
| 140 |
text_gen_pipeline = pipeline(
|
| 141 |
"text-generation",
|