Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,8 @@ import spaces
|
|
2 |
import gradio as gr
|
3 |
from cartesia_pytorch import ReneLMHeadModel
|
4 |
from transformers import AutoTokenizer
|
5 |
-
import subprocess
|
6 |
-
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
7 |
# Load model and tokenizer
|
8 |
model = ReneLMHeadModel.from_pretrained("cartesia-ai/Rene-v0.1-1.3b-pytorch").half().cuda()
|
9 |
tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
|
|
|
2 |
import gradio as gr
|
3 |
from cartesia_pytorch import ReneLMHeadModel
|
4 |
from transformers import AutoTokenizer
|
5 |
+
#import subprocess
|
6 |
+
#subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
7 |
# Load model and tokenizer
|
8 |
model = ReneLMHeadModel.from_pretrained("cartesia-ai/Rene-v0.1-1.3b-pytorch").half().cuda()
|
9 |
tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
|