Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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)
|