AC2513 commited on
Commit
ce64ebd
·
1 Parent(s): d54c706

update torch compilation

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  from huggingface_hub import login
3
  from collections.abc import Iterator
4
  from transformers import (
@@ -6,7 +7,6 @@ from transformers import (
6
  TextIteratorStreamer,
7
  Gemma3Processor,
8
  Gemma3nForConditionalGeneration,
9
- Gemma3ForCausalLM
10
  )
11
  import spaces
12
  import tempfile
 
1
  import torch
2
+ torch._dynamo.config.disable = True
3
  from huggingface_hub import login
4
  from collections.abc import Iterator
5
  from transformers import (
 
7
  TextIteratorStreamer,
8
  Gemma3Processor,
9
  Gemma3nForConditionalGeneration,
 
10
  )
11
  import spaces
12
  import tempfile