rinrikatoki commited on
Commit
ea2466c
·
verified ·
1 Parent(s): fa412e4

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
6
  from peft import PeftModel
7
  from huggingface_hub import login
8
 
 
9
  # --- گام ۱: احراز هویت Hugging Face
10
  hf_token = os.environ.get("HF_TOKEN")
11
  if not hf_token:
@@ -32,7 +33,7 @@ tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, use_auth_token=hf_token)
32
  base_model = AutoModelForCausalLM.from_pretrained(
33
  BASE_MODEL,
34
  device_map="auto",
35
- token=HF_TOKEN,
36
  trust_remote_code=True
37
  )
38
 
 
6
  from peft import PeftModel
7
  from huggingface_hub import login
8
 
9
+
10
  # --- گام ۱: احراز هویت Hugging Face
11
  hf_token = os.environ.get("HF_TOKEN")
12
  if not hf_token:
 
33
  base_model = AutoModelForCausalLM.from_pretrained(
34
  BASE_MODEL,
35
  device_map="auto",
36
+ token=hf_token,
37
  trust_remote_code=True
38
  )
39