Spaces:

LymphSteamer
/

low_haiku_space

Running

LymphSteamer commited on May 31

Commit

da9043e

verified ·

1 Parent(s): 307cab1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCausalLM,BitsAndBytesConfig
 import torch
 model_id = "cyberagent/open-calm-1b"
-tokenizer = AutoTokenizer.from_pretrained(model_id,use_fast=False,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map="auto",

 from transformers import AutoTokenizer,AutoModelForCausalLM,BitsAndBytesConfig
 import torch
 model_id = "cyberagent/open-calm-1b"
+tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map="auto",