asahi417 commited on
Commit
92d2f97
·
verified ·
1 Parent(s): 1ef6a32

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
 
1
  import gradio as gr
2
  import spaces
3
  from panna import Flux1Dev
4
 
5
 
6
- model = Flux1Dev()
7
  title = ("# [Flux 1 Dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)\n"
8
  "The demo is part of [panna](https://github.com/asahi417/panna) project.")
9
  examples = [
 
1
+ import torch
2
  import gradio as gr
3
  import spaces
4
  from panna import Flux1Dev
5
 
6
 
7
+ model = Flux1Dev(torch_dtype=torch.bfloat16)
8
  title = ("# [Flux 1 Dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)\n"
9
  "The demo is part of [panna](https://github.com/asahi417/panna) project.")
10
  examples = [