srinuksv commited on
Commit
4887e87
Β·
verified Β·
1 Parent(s): cc17f44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ import soundfile as sf
9
  import modelscope_studio.components.base as ms
10
  import modelscope_studio.components.antd as antd
11
  import gradio.processing_utils as processing_utils
12
-
 
13
  from transformers import Qwen2_5OmniModel, Qwen2_5OmniProcessor
14
  from gradio_client import utils as client_utils
15
  from qwen_omni_utils import process_mm_info
 
9
  import modelscope_studio.components.base as ms
10
  import modelscope_studio.components.antd as antd
11
  import gradio.processing_utils as processing_utils
12
+ from transformers import AutoModelForCausalLM
13
+ from accelerate import disk_offload
14
  from transformers import Qwen2_5OmniModel, Qwen2_5OmniProcessor
15
  from gradio_client import utils as client_utils
16
  from qwen_omni_utils import process_mm_info