Spaces:

tttoaster
/

SEED-X-17B

Build error

tttoaster commited on May 5, 2024

Commit

b1eafeb

verified ·

1 Parent(s): 3e51c05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,8 +24,8 @@ print("device", device)
 print("PyTorch version:", torch.__version__)
 print("CUDA version:", torch.version.cuda)
-import os
 os.environ['CUDA_VISIBLE_DEVICE']='0'
 subprocess.call(['python3', 'src/demo/seed_llama_flask.py', '--image_transform', 'configs/processer/qwen_448_transform.yaml', \
                       '--tokenizer', 'configs/tokenizer/clm_llama_tokenizer_224loc_anyres.yaml', \

 print("PyTorch version:", torch.__version__)
 print("CUDA version:", torch.version.cuda)
 os.environ['CUDA_VISIBLE_DEVICE']='0'
+os.environ["MKL_THREADING_LAYER"] = "GNU"
 subprocess.call(['python3', 'src/demo/seed_llama_flask.py', '--image_transform', 'configs/processer/qwen_448_transform.yaml', \
                       '--tokenizer', 'configs/tokenizer/clm_llama_tokenizer_224loc_anyres.yaml', \