Spaces:

CreitinGameplays
/

ConvAIChat

Runtime error

CreitinGameplays commited on May 19, 2024

Commit

c6eb853

verified ·

1 Parent(s): 4fbfff8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,6 @@ if torch.cuda.is_available():
     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 @spaces.GPU
 def generate(
     message: str,

     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 @spaces.GPU
 def generate(
     message: str,