laserbeam2045 commited on
Commit
a9bf179
·
1 Parent(s): 1f50e47
Files changed (2) hide show
  1. app.py +5 -3
  2. requirements.txt +2 -0
app.py CHANGED
@@ -22,13 +22,15 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
  # -----------------------------------------------------------------------------
23
  tokenizer = AutoTokenizer.from_pretrained(
24
  MODEL_ID,
25
- use_auth_token=HF_TOKEN,
26
- trust_remote_code=True
 
 
27
  )
28
 
29
  model = AutoModelForCausalLM.from_pretrained(
30
  MODEL_ID,
31
- use_auth_token=HF_TOKEN,
32
  torch_dtype=torch.float32, # CPU 環境では float32
33
  device_map="auto" if torch.cuda.is_available() else None
34
  )
 
22
  # -----------------------------------------------------------------------------
23
  tokenizer = AutoTokenizer.from_pretrained(
24
  MODEL_ID,
25
+ token=HF_TOKEN,
26
+ trust_remote_code=True,
27
+ torch_dtype=torch.float32,
28
+ low_cpu_mem_usage=True
29
  )
30
 
31
  model = AutoModelForCausalLM.from_pretrained(
32
  MODEL_ID,
33
+ token=HF_TOKEN,
34
  torch_dtype=torch.float32, # CPU 環境では float32
35
  device_map="auto" if torch.cuda.is_available() else None
36
  )
requirements.txt CHANGED
@@ -2,3 +2,5 @@ fastapi
2
  uvicorn[standard]
3
  transformers>=4.50.0.dev0
4
  torch
 
 
 
2
  uvicorn[standard]
3
  transformers>=4.50.0.dev0
4
  torch
5
+ accelerate>=0.9.0
6
+ safetensors