Hugging Face's logo

Spaces:
VanguardAI
/
BhashiniSpace_Text_gemma
Runtime error

App Files Community
FlawedLLM commited on
Commit
727f0da
•
1 Parent(s): 99c292a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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",use_safetensors= True ,)
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):