📝

HamedBabaei · HamedBabaei · commit ca3ebf94bb36 · 2025-10-01T17:05:06.000+02:00
diff --git a/docs/source/learners/llm.rst b/docs/source/learners/llm.rst
@@ -267,7 +267,7 @@ Once your custom class is defined, you can pass it into ``AutoLLMLearner``:
         device="cuda"
     )
 
-    falcon_learner.llm.load(model_id="tiiuae/falcon-7b-instruct")
+    falcon_learner.llm.load(model_id="tiiuae/Falcon-H1-1.5B-Deep-Instruct")
 
     # Train and evaluate
     falcon_learner.fit(train_data, task="term-typing")

Original file line number	Diff line number	Diff line change
@@ -267,7 +267,7 @@ Once your custom class is defined, you can pass it into ``AutoLLMLearner``:
`267`	`267`	`device="cuda"`
`268`	`268`	`)`
`269`	`269`
`270`		`- falcon_learner.llm.load(model_id="tiiuae/falcon-7b-instruct")`
	`270`	`+ falcon_learner.llm.load(model_id="tiiuae/Falcon-H1-1.5B-Deep-Instruct")`
`271`	`271`
`272`	`272`	`# Train and evaluate`
`273`	`273`	`falcon_learner.fit(train_data, task="term-typing")`