BhashiniSpace_Text_gemma

Runtime error

FlawedLLM commited on May 23

Commit

17285b7

•

1 Parent(s): 067d452

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,6 @@ model = AutoModelForCausalLM.from_pretrained("FlawedLLM/Bhashini_gemma_merged16b
                                              quantization_config=quantization_config,
                                              torch_dtype =torch.float16,
                                              low_cpu_mem_usage=True,
-                                             use_safetensors=True,
                                              trust_remote_code=True)
 # alpaca_prompt = You MUST copy from above!

                                              quantization_config=quantization_config,
                                              torch_dtype =torch.float16,
                                              low_cpu_mem_usage=True,
                                              trust_remote_code=True)
 # alpaca_prompt = You MUST copy from above!