Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from threading import Thread
|
|
3 |
from typing import Iterator
|
4 |
|
5 |
import gradio as gr
|
6 |
-
import spaces
|
7 |
import torch
|
8 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
9 |
from peft import AutoPeftModelForCausalLM
|
@@ -52,7 +52,7 @@ api.upload_folder(
|
|
52 |
)
|
53 |
|
54 |
|
55 |
-
|
56 |
def generate(
|
57 |
message: str,
|
58 |
chat_history: list[dict],
|
|
|
3 |
from typing import Iterator
|
4 |
|
5 |
import gradio as gr
|
6 |
+
#import spaces
|
7 |
import torch
|
8 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
9 |
from peft import AutoPeftModelForCausalLM
|
|
|
52 |
)
|
53 |
|
54 |
|
55 |
+
#@spaces.GPU()
|
56 |
def generate(
|
57 |
message: str,
|
58 |
chat_history: list[dict],
|