red-kit
/

loop-gpt-image-handler

redkits commited on Aug 15

Commit

2ca5d8d

verified ·

1 Parent(s): a9ec93e

Upload server.py with huggingface_hub

Files changed (1) hide show

server.py CHANGED Viewed

@@ -5,8 +5,8 @@ import base64
 import io
 import json
-# Install diffusers from source (supports GLM-Image) + flask for serving
-subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", "git+https://github.com/huggingface/diffusers.git", "flask", "accelerate", "sentencepiece", "protobuf"])
 import torch
 from flask import Flask, request, jsonify

 import io
 import json
+# Install diffusers (latest PyPI supports GLM-Image) + flask for serving
+subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", "--upgrade", "diffusers", "flask", "accelerate", "sentencepiece", "protobuf", "transformers"])
 import torch
 from flask import Flask, request, jsonify