digitalassistant-ai commited on
Commit
6abb245
·
verified ·
1 Parent(s): 18386ea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -51,10 +51,10 @@ pip install transformers accelerate torch --upgrade
51
 
52
  Minimal inference (local path):
53
 
54
- ```python
55
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
56
 
57
- MODEL_PATH = "https://huggingface.co/wwwywcom/Selling-Assistant-V1/tree/main/model"
58
 
59
  tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
60
  model = AutoModelForCausalLM.from_pretrained(MODEL_PATH, device_map="auto", torch_dtype="auto")
 
51
 
52
  Minimal inference (local path):
53
 
54
+ ```python
55
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
56
 
57
+ MODEL_PATH = "https://huggingface.co/digitalassistant-ai/Selling-Assistant-V1/model"
58
 
59
  tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
60
  model = AutoModelForCausalLM.from_pretrained(MODEL_PATH, device_map="auto", torch_dtype="auto")