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")
|