Spaces:
Runtime error
Runtime error
FlawedLLM
commited on
Commit
•
727f0da
1
Parent(s):
99c292a
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ tokenizer = AutoTokenizer.from_pretrained("FlawedLLM/Bhashini_9")
|
|
54 |
# )
|
55 |
|
56 |
# torch_dtype =torch.float16
|
57 |
-
model = AutoModelForCausalLM.from_pretrained("FlawedLLM/Bhashini_9",
|
58 |
|
59 |
@spaces.GPU(duration=300)
|
60 |
def chunk_it(input_command):
|
|
|
54 |
# )
|
55 |
|
56 |
# torch_dtype =torch.float16
|
57 |
+
model = AutoModelForCausalLM.from_pretrained("FlawedLLM/Bhashini_9",ignore_mismatched_sizes=True).to('cuda')
|
58 |
|
59 |
@spaces.GPU(duration=300)
|
60 |
def chunk_it(input_command):
|