Shago commited on
Commit
c0ea8ea
·
verified ·
1 Parent(s): 0d57faa

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +1 -1
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-litert-preview",
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",