Spaces:

LymphSteamer
/

low_haiku_space

Sleeping

LymphSteamer commited on May 31

Commit

808c9d5

verified ·

1 Parent(s): 5f84da7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCausalLM
 import torch
-model_id = "rinna/japanese-gpt2-small"
 tokenizer = AutoTokenizer.from_pretrained(model_id,use_fast=False,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,

 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCausalLM
 import torch
+model_id = "mlx-community/llm-jp-3-1.8b-instruct3-4bit"
 tokenizer = AutoTokenizer.from_pretrained(model_id,use_fast=False,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,