alexkueck commited on
Commit
2a0e527
1 Parent(s): d7b7c3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -71,18 +71,18 @@ splittet = False
71
  # inference client
72
  ##############################################
73
  print ("Inf.Client")
74
- #client = InferenceClient("https://api-inference.huggingface.co/models/meta-llama/Llama-2-70b-chat-hf")
75
- client = InferenceClient("https://ybdhvwle4ksrawzo.eu-west-1.aws.endpoints.huggingface.cloud")
76
  #client = InferenceClient(model="TheBloke/Yi-34B-Chat-GGUF")
77
 
78
  ##############################################
79
  # tokenizer for generating prompt
80
  ##############################################
81
  print ("Tokenizer")
82
- #tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-70b-chat-hf")
83
  #tokenizer = AutoTokenizer.from_pretrained("TheBloke/Yi-34B-Chat-GGUF")
84
  #tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
85
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
86
 
87
  ##############################################
88
  # Zum Testen:
 
71
  # inference client
72
  ##############################################
73
  print ("Inf.Client")
74
+ client = InferenceClient("https://api-inference.huggingface.co/models/meta-llama/Llama-2-70b-chat-hf")
75
+ #client = InferenceClient("https://ybdhvwle4ksrawzo.eu-west-1.aws.endpoints.huggingface.cloud")
76
  #client = InferenceClient(model="TheBloke/Yi-34B-Chat-GGUF")
77
 
78
  ##############################################
79
  # tokenizer for generating prompt
80
  ##############################################
81
  print ("Tokenizer")
82
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-70b-chat-hf")
83
  #tokenizer = AutoTokenizer.from_pretrained("TheBloke/Yi-34B-Chat-GGUF")
84
  #tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
85
+ #tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
86
 
87
  ##############################################
88
  # Zum Testen: