Spaces:
Sleeping
Sleeping
Update llm.py
Browse files
llm.py
CHANGED
|
@@ -7,7 +7,7 @@ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
|
|
| 7 |
# Initialize HF pipeline for text generation
|
| 8 |
text_generator = pipeline(
|
| 9 |
"text-generation", # Task type
|
| 10 |
-
model="google/gemma-3n-E2B-it
|
| 11 |
# model="google/gemma-3n-e4b-it",
|
| 12 |
# model="Qwen/Qwen3-Embedding-0.6B",
|
| 13 |
# device="cuda" if torch.cuda.is_available() else "cpu",
|
|
|
|
| 7 |
# Initialize HF pipeline for text generation
|
| 8 |
text_generator = pipeline(
|
| 9 |
"text-generation", # Task type
|
| 10 |
+
model="google/gemma-3n-E2B-it",
|
| 11 |
# model="google/gemma-3n-e4b-it",
|
| 12 |
# model="Qwen/Qwen3-Embedding-0.6B",
|
| 13 |
# device="cuda" if torch.cuda.is_available() else "cpu",
|