Spaces:
Runtime error
Runtime error
FlawedLLM
commited on
Commit
•
52c65e6
1
Parent(s):
731e7a8
Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
62 |
# Load model directly
|
63 |
|
64 |
tokenizer = AutoTokenizer.from_pretrained("FlawedLLM/Bhashini_00")
|
65 |
-
model = AutoModelForCausalLM.from_pretrained("FlawedLLM/Bhashini_00", load_in_4bit=True)
|
66 |
|
67 |
@spaces.GPU(duration=300)
|
68 |
def chunk_it(input_command):
|
|
|
62 |
# Load model directly
|
63 |
|
64 |
tokenizer = AutoTokenizer.from_pretrained("FlawedLLM/Bhashini_00")
|
65 |
+
model = AutoModelForCausalLM.from_pretrained("FlawedLLM/Bhashini_00", load_in_4bit=True)
|
66 |
|
67 |
@spaces.GPU(duration=300)
|
68 |
def chunk_it(input_command):
|