Lord-Raven commited on
Commit
b875571
·
1 Parent(s): d65667c

Messing with configuration.

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -36,13 +36,13 @@ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
36
  # model = ORTModelForSequenceClassification.from_pretrained(model_name, export=True, provider="CUDAExecutionProvider")
37
  # tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
- # model = ORTModelForSequenceClassification.from_pretrained(
40
- # "philschmid/tiny-bert-sst2-distilled",
41
- # export=True,
42
- # provider="CUDAExecutionProvider",
43
- # )
44
 
45
- # tokenizer = AutoTokenizer.from_pretrained("philschmid/tiny-bert-sst2-distilled")
46
 
47
  # classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer, device="cuda:0")
48
 
 
36
  # model = ORTModelForSequenceClassification.from_pretrained(model_name, export=True, provider="CUDAExecutionProvider")
37
  # tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
+ model = ORTModelForSequenceClassification.from_pretrained(
40
+ "philschmid/tiny-bert-sst2-distilled",
41
+ export=True,
42
+ provider="CUDAExecutionProvider",
43
+ )
44
 
45
+ tokenizer = AutoTokenizer.from_pretrained("philschmid/tiny-bert-sst2-distilled")
46
 
47
  # classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer, device="cuda:0")
48