Yurii Paniv commited on
Commit
08127ab
·
1 Parent(s): 6e047fb

Lower limit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,12 +16,12 @@ quant_config = BitsAndBytesConfig(
16
  )
17
 
18
  model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1",
19
- quantization_config=quant_config,).to("cuda")
20
  #device_map="auto",)
21
  model = PeftModel.from_pretrained(model, "lang-uk/dragoman")
22
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False, add_bos_token=False)
23
 
24
- @spaces.GPU(duration=200)
25
  def translate(input_text):
26
  # iteratively generate
27
  input_text = input_text.strip()
 
16
  )
17
 
18
  model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1",
19
+ quantization_config=quant_config,)#.to("cuda")
20
  #device_map="auto",)
21
  model = PeftModel.from_pretrained(model, "lang-uk/dragoman")
22
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False, add_bos_token=False)
23
 
24
+ @spaces.GPU(duration=20)
25
  def translate(input_text):
26
  # iteratively generate
27
  input_text = input_text.strip()