yongyeol commited on
Commit
4ae4657
Β·
verified Β·
1 Parent(s): a71d9a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -8
app.py CHANGED
@@ -1,19 +1,26 @@
1
 
2
- import os, subprocess, sys
3
- os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1" # safetensors
4
- os.environ["XFORMERS_FORCE_DISABLE"] = "1" # βœ… xformers μ™„μ „ λΉ„ν™œμ„±ν™”
5
 
 
 
 
 
 
 
6
 
7
- # ─── audiocraft 동적 μ„€μΉ˜ (ν•œ 번만) ───
8
  try:
9
  from audiocraft.models import MusicGen
10
  except ModuleNotFoundError:
11
- subprocess.check_call(
12
- [sys.executable, "-m", "pip", "install",
13
- "git+https://github.com/facebookresearch/audiocraft@main", "--no-deps"]
14
- )
15
  from audiocraft.models import MusicGen
16
 
 
17
  import gradio as gr
18
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
19
  from audiocraft.models import MusicGen
 
1
 
2
+ import os, sys, types, subprocess
3
+ os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1"
4
+ os.environ["XFORMERS_FORCE_DISABLE"] = "1" # κ·Έλž˜λ„ importλŠ” λ°œμƒ β†’ κ°€μ§œ λͺ¨λ“ˆ μ£Όμž…
5
 
6
+ # ── ✨ xformers 더미 λͺ¨λ“ˆ μ‚½μž… ─────────────────────────
7
+ dummy = types.ModuleType("xformers")
8
+ dummy.ops = types.ModuleType("xformers.ops") # audiocraftκ°€ ops ν•˜μœ„λͺ¨λ“ˆλ„ 찾음
9
+ sys.modules["xformers"] = dummy
10
+ sys.modules["xformers.ops"] = dummy.ops
11
+ # ─────────────────────────────────────────────────────
12
 
13
+ # ── audiocraft 동적 μ„€μΉ˜ ──
14
  try:
15
  from audiocraft.models import MusicGen
16
  except ModuleNotFoundError:
17
+ subprocess.check_call([
18
+ sys.executable, "-m", "pip", "install",
19
+ "git+https://github.com/facebookresearch/audiocraft@main", "--no-deps"
20
+ ])
21
  from audiocraft.models import MusicGen
22
 
23
+
24
  import gradio as gr
25
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
26
  from audiocraft.models import MusicGen