tangzhy commited on
Commit
d3128ed
1 Parent(s): 1579796

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ model_id = "CardinalOperations/ORLM-LLaMA-3-8B"
46
  # model.eval()
47
 
48
  subprocess.run(f'huggingface-cli download {model_id} --local-dir ./local_model', shell=True)
49
- model = LLM(model='./local_model', tensor_parallel_size=torch.cuda.device_count())
50
  print("init model done.")
51
 
52
  @spaces.GPU(duration=60)
 
46
  # model.eval()
47
 
48
  subprocess.run(f'huggingface-cli download {model_id} --local-dir ./local_model', shell=True)
49
+ model = LLM(model='./local_model', tensor_parallel_size=1)
50
  print("init model done.")
51
 
52
  @spaces.GPU(duration=60)