BioTATA-7B

Runtime error

kimou605 commited on May 5

Commit

d546413

•

1 Parent(s): 3518906

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from transformers import pipeline
-pipe = pipeline("text-generation", model="kimou605/shadow-clown-BioMistral-7B-DARE", torch_dtype=torch.bfloat16, device_map="auto")
 @spaces.GPU

 import torch
 from transformers import pipeline
+pipe = pipeline("text-generation", model="kimou605/BioTATA-7B", torch_dtype=torch.bfloat16, device_map="auto")
 @spaces.GPU