Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ description = """ try this space to build downstream applications with [CohereFo
|
|
9 |
device = "cuda"
|
10 |
checkpoint = "CohereForAI/aya-101"
|
11 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
12 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint, torch_dtype=torch.float16, low_cpu_mem_usage=True, device_map=
|
13 |
|
14 |
@spaces.GPU
|
15 |
def aya(text):
|
|
|
9 |
device = "cuda"
|
10 |
checkpoint = "CohereForAI/aya-101"
|
11 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
12 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint, torch_dtype=torch.float16, low_cpu_mem_usage=True, device_map=device)
|
13 |
|
14 |
@spaces.GPU
|
15 |
def aya(text):
|