mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-03-11 21:07:20 +00:00
LangChain WIP; TypeError: _BaseAutoModelClass.from_pretrained() missing 1 required positional argument: 'pretrained_model_name_or_path'
This commit is contained in:
@@ -17,7 +17,7 @@ class Llm:
|
||||
model_path = os.path.join(base_dir, "phi3")
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
model_id=model_path,
|
||||
pretrained_model_name_or_path=model_path,
|
||||
device_map="cpu", # Use available GPU
|
||||
trust_remote_code=True, # If model requires custom code
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user